def getuid(username): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("select UID from user where UserName = '******';" %(username)) uid = cursor.fetchall()[0][0] db.close() return uid
def reqState(username): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("""select State from user where UserName = '******'; """ % (username)) status = cursor.fetchall()[0][0] db.close() return status
def outState(username): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("""update user set State = 0 where UserName = '******'; """ % (username)) db.commit() db.close() return
def logIn(username): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("""select Password from user where UserName ='******'; """ % (username)) pw = cursor.fetchall()[0][0] db.close() return pw
def dele(username, sid): UID =getuid(username) SID = sid db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("delete from preferlist where SID = '%s' and UID = '%s';" %(SID,UID)) db.commit() db.close() return
def stockList(): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute(""" select * from stock ;""") all = cursor.fetchall() data = '[' for i in all: data = data + str(i[0]) + ',' data = data[:-1] + ']' return data
def allData(sid): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") name = getTableName(sid) cursor.execute(""" select * from %s ;""" % (name)) all = cursor.fetchall() data = '' for i in all: data = data + str(i[0]) + ':' + str(i[1]) + ',' return data[:-1]
def periodData(timeStart, sid): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") name = getTableName(sid) cursor.execute("""select * from %s where times=%d;""" % (name, timeStart)) period = cursor.fetchall() data = '' for i in period: data = data + str(i[0]) + ':' + str(i[1]) + ',' return data[:-1]
def checkexist(username): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("""select count(*) from user where UserName = '******';""" % (username)) r = cursor.fetchall()[0][0] db.close() if r == 0: return 0 else: return 1
def req(username): UID = getuid(username) db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("select SID from preferlist where UID = '%s'; " %(UID)) d = cursor.fetchall() plist ='[' if d != (): for i in d: plist = plist +i[0]+',' plist = plist[:-1] +']' return plist
def up(username, password): db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("select count(*) from user;") num = cursor.fetchall()[0][0] id = num + 1 uid = str(id) cursor.execute("""insert into user values('%s', '%s', '%s', default);""" % (uid, username, password)) db.commit() db.close() return
def add(username, sid): UID = getuid(username) SID = sid db = connect.connect() cursor = db.cursor() cursor.execute("use stock_system;") cursor.execute("select count(*) from preferlist where SID = '%s' and UID = '%s';" %(SID,UID)) count = cursor.fetchall()[0][0] if count > 0: db.close() return else: cursor.execute("insert into preferlist values('%s', '%s');" %(UID,SID)) db.commit() db.close() return