def setPort(self, port): sql = """UPDATE Users SET port='%(port)s' WHERE login = '******' """%{"l":self.login,"port":port} d = DB('localhost','root','messenger','root') d.execute(sql)
def active_kuaidi_tixing(cid): try: db = DB() sql = "update t_dd_kuaidi set active=1 where cid = '%s'" % cid db.update(sql) print "Successful to Active cid:%s from t_dd_kuaidi!" % cid except Exception,e: print "Error:Faild active cid:%s from t_dd_kuaidi!" % cid print "Failed info:", e print "sql:",sql
def response_ding_action(sid, cid, push_token): db = DB() sql = "select com,nu from t_dd_kuaidi where cid='%s'" % cid data = db.query(sql) com = data[0][0] nu = data[0][1] dtype = "ding" q.enqueue(query_kuaidi_info, sid, cid, com, nu, dtype, push_token)
def reg(self, p): self.password = hex(GOST(256).hash(p.encode('hex') + self.login.encode('hex'))) self.__pswd = hex(GOST(256).hash(self.password.encode('hex')))[:32] sql = """INSERT INTO Users(login, passwd) VALUES ('%(l)s', '%(p)s') """%{"l":self.login, "p": self.password} d = DB('localhost','root','messenger','root') d.execute(sql) self.genRSAKey() self.genDSKey() return self.password
async def echo(websocket, path): async for message in websocket: global db db = DB() if (message == "recent"): await websocket.send(encodeJSONTotal(db.get_all_weekly())) elif ("daily" in message): await websocket.send(encodeJSON(daily(message[6:]))) elif ("weekly" in message): await websocket.send(encodeJSON(weekly(message[7:]))) elif ("monthly" in message): await websocket.send(encodeJSON(monthly(message[8:]))) db.close()
def decriptRSAKey(self): sql = """SELECT priv, iv from RSA WHERE login='******' """%{"l":self.login} d = DB('localhost','root','messenger','root') key, iv = d.execute(sql) key = key.decode('hex') iv = iv.decode('hex') cipher = AES.new(self.__pswd, AES.MODE_CFB, iv) key = cipher.decrypt(key) self.__PrivRSAKey = RSA.importKey(key)
def loop_kuaidi_task(sid): db = DB() sql = """ select cid, com, nu from t_dd_kuaidi where active=1 """ print "Start loop kaudi task..." data = db.query(sql) dtype = 'loop' for record in data: cid, com, nu = record[0], record[1], record[2] q.enqueue(query_kuaidi_info, sid, cid, com, nu, dtype)
def genRSAKey(self): key = RSA.generate(self.RSALenghtKey) pub = key.publickey().exportKey() priv = key.exportKey() AES.block_size = 32 iv = os.urandom(8).encode('hex') # Encryption cipher = AES.new(self.__pswd, AES.MODE_CFB, iv) priv = cipher.encrypt(priv) sql = """INSERT INTO RSA(login, pub, priv, iv) VALUES ('%(l)s', '%(pub)s', '%(priv)s', '%(iv)s') """%{"l":self.login,"pub":pub.encode('hex'), "priv":priv.encode('hex'), "iv":iv.encode('hex')} d = DB('localhost','root','messenger','root') d.execute(sql)
def checkSign(self, login): sql = """SELECT x, y from DS WHERE login='******' """%{"l":login} d = DB('localhost','root','messenger','root') E = d.execute(sql) E = map(lambda h: int(h[2:-1], 16), E) Q = ECPoint(E[0], E[1], a, b, p) return DS.SingVer(H, sign, Q)
def add_kuaidi_tixing(cid, com, nu): db = DB() now = int(time.time()) create_time = datetime.fromtimestamp(now) try: sql = "insert into t_dd_kuaidi(cid, com, nu, create_time, active)\ values('{0}', '{1}', '{2}', '{3}', 1)".format(cid, com, nu, create_time) db.update(sql) print "Successful to add kuaidi tixing: cid=%s, com=%s,\ nu=%s" % (cid, com, nu) except Exception,e: print "Error: Failed to add kuaidi tixing: cid=%s, com=%s,\ nu=%s" % (cid, com, nu) print "Failed info:", e print "sql:", sql
def genDSKey(self): priv = self.DS.GenPrivateKey() pub = self.DS.GenPublicKey(priv) pub = pub.getAttr() AES.block_size = 32 iv = os.urandom(8).encode('hex') # Encryption cipher = AES.new(self.__pswd, AES.MODE_CFB, iv) priv = cipher.encrypt(hex(priv)) sql = """INSERT INTO DS(login, x, y, a, b, fieldchar, priv, iv) VALUES ('%(l)s', '%(x)s', '%(y)s', '%(a)s', '%(b)s', '%(f)s', '%(priv)s', '%(iv)s') """%{"l":self.login,"x":hex(pub[0]), "y":hex(pub[1]), "a":hex(pub[2]), "b":hex(pub[3]), "f":hex(pub[4]), "priv":priv.encode('hex'), "iv":iv.encode('hex')} d = DB('localhost','root','messenger','root') d.execute(sql)
def updateRSAKey(self): key = RSA.generate(self.RSALenghtKey) pub = key.publickey().exportKey() priv = key.exportKey() AES.block_size = 32 iv = os.urandom(8).encode('hex') # Encryption cipher = AES.new(self.__pswd, AES.MODE_CFB, iv) priv = cipher.encrypt(priv) sql = """UPDATE RSA SET pub='%(pub)s', priv='%(priv)s', iv='%(iv)s' WHERE login = '******' """%{"l":self.login,"pub":pub.encode('hex'), "priv":priv.encode('hex'), "iv":iv.encode('hex')} d = DB('localhost','root','messenger','root') d.execute(sql)
def updateDSKey(self): priv = self.DS.GenPrivateKey() pub = self.DS.GenPublicKey(priv) pub = pub.getAttr() AES.block_size = 32 iv = os.urandom(8).encode('hex') # Encryption cipher = AES.new(self.__pswd, AES.MODE_CFB, iv) priv = cipher.encrypt(hex(priv)) sql = """UPDATE DS SET x='%(x)s', y ='%(y)s', priv='%(priv)s', iv='%(iv)s' WHERE login = '******' """%{"x":hex(pub[0]), "y":hex(pub[1]), "priv":priv.encode('hex'), "iv":iv.encode('hex')} d = DB('localhost','root','messenger','root') d.execute(sql)
def __init__(self, login, port): self.login = login self.port = port sql = """SELECT passwd from Users WHERE login='******' """%{"l":self.login} d = DB('localhost','root','messenger','root') q = d.execute(sql) print q if q is None: self.password = q else: self.password = q[0] self.__pswd = hex(GOST(256).hash(self.password.encode('hex')))[:32] self.signature()
def encryptMsg(self, login, msg): sql = """SELECT pub from RSA WHERE login='******' """%{"l":login} d = DB('localhost','root','messenger','root') key = d.execute(sql)[0] key = key.decode('hex') PubKey = RSA.importKey(key) iv = os.urandom(4).encode('hex') H = GOST(256).hash(msg.encode('hex')) print self.__PrivDSKey self.DS.GenPublicKey(self.__PrivDSKey,self.b) sign = self.DS.SingGen(32, self.__PrivDSKey) pack = "#---#".join(msg, sign) print pack cipher_text = ''.join(PubKey.encrypt(pack, iv)).encode('hex') print 'messenge encrypt' return
def push_handler(sid, cid, ck_time, last_time, context, dtype, push_token): db = DB() sql = "select state_time, check_time from t_dd_kuaidi where cid='%s'" % cid data = db.query(sql) state_time = data[0][0] check_time = data[0][1] # update check_time udb = DB() sql = "update t_dd_kuaidi set check_time='%s' where cid='%s'" % (ck_time, cid) print "Update check_time,sql:", sql udb.update(sql) #print "state_time:%s, last_time:%s" % (state_time, last_time) if state_time != last_time or dtype=='ding': # push new state and update push_time, state_time print "Start push message..." Push = PushService(sid, cid, push_token) status_code = Push.push_text('快递提醒', context) now = int(time.time()) push_time = datetime.fromtimestamp(now) if status_code == 200: print "Push message Successful!" sdb = DB() sql = "update t_dd_kuaidi set state_time='%s',\ push_time='%s' where cid='%s'" % (last_time, push_time, cid) sdb.update(sql) print "Update state_time, push_time:%s" % sql else: print "Push message Error!" print "Status code: %s, Error accurs when push new state: cid=%s, ck_time=%s,\ last_time=%s, context=%s, push_token=%s, push_time=%s\ " % (status_code, cid, ck_time, last_time, context, push_token, push_time)
from MySQL import DB from Artista import Artista c = DB.run("select * from Artista") for b in c: unArtista = Artista unArtista.idArtist = b['idArtista'] unArtista.Nombre = b['Nombre'] unArtista.Apellido = b['Apellido'] unArtista.Conciertos = b['Conciertos']
import asyncio import websockets from MySQL import DB import json import datetime db = DB() #this is all terrible old code and needs to be severly rewritten for a larger scale project :/ def encodeJSONTotal(data): JSONData = {} for i in range(len(data)): if JSONData.get(data[i][0]) != None: JSONData[data[i][0]]['minutes'] += 1 else: JSONData[data[i][0]] = {'minutes': 1} return json.dumps(JSONData) def encodeJSON(data): return json.dumps(data) def daily(name): info = db.get_user_daily(name) day = {} for k in info: if day.get(k[1].hour): day[k[1].hour] += 1 else:
def main(): database = DB() repository = Repository(database) controller = Controller(repository) view = View(controller) view.mainMenu()
return msg def setPort(self, port): sql = """UPDATE Users SET port='%(port)s' WHERE login = '******' """%{"l":self.login,"port":port} d = DB('localhost','root','messenger','root') d.execute(sql) if __name__ == '__main__': login = '******' password = '******' password = hex(GOST(256).hash(password.encode('hex') + login.encode('hex'))) sql = """INSERT INTO Users(login, passwd) VALUES ('%(l)s', '%(p)s') """%{"l":login, "p": password} d = DB('localhost','root','messenger','root') d.execute(sql) #Q = newUser.genRSAKey() #newUser.genRSAKey() #newUser.decriptDSKey() #print Q #print Q.decode('hex')
def SetInsert(self, idArtista, Nombre, Apellido, Conciertos): DB.run("insert into Artista values (null,'" + self.Nombre + "','" + self.Apellido + "')")