Skip to main content

Postgresql program

Postgresql program 

#!python

import psycopg2

class KKMsgDao:
     def __init__(self):
          pass

     def getMsg(self):
          ret = []
          db = psycopg2.connect(host='localhost', dbname='kkrack',user='xxxxxxx',password='xxxxxxxx',port=5432) 
          curs = db.cursor()
          sql = "SELECT * FROM KKRACK ORDER BY TIME DESC LIMIT 5;"
          curs.execute(sql)
          rows = curs.fetchall()
          for e in rows:
               temp = {'username':e[0],'message':e[1],'time':e[2],'etc':e[3] }
               ret.append(temp)
          db.commit()
          db.close()
          return ret


     def insMsg(self, username, message, clientIP):
          db = psycopg2.connect(host='localhost', dbname='kkrack',user='xxxxxxxxx',password='xxxxxx',port=5432) 
          curs = db.cursor()
          sql = '''insert into KKRACK (username, message, time, etc) values(%s,%s, NOW(), %s)'''
          curs.execute(sql,(username, message, clientIP))
          db.commit()
          db.close()

if __name__ =='__main__':
     kkMsgDao = KKMsgDao()
     msg = kkMsgDao.getMsg()
     print(msg)
     #kkMsgDao.insEmp("hyunsu", "hahaha", "10.10.10.1")