def change_pwd(username,pwd): mydb.open() result=_valid_login_format(username,pwd) if not result==None: return result mydb.insert_db("""update account set pwd=? WHERE username= ?""", [pwd, username]) mydb.close_db() return None
def add_connection(nid1,nid2): mydb.open() p1 = mydb.query_db('SELECT * FROM node WHERE nid = ?', [nid1], one = True) p2 = mydb.query_db('SELECT * FROM node WHERE nid = ?', [nid2], one = True) mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)', [p1['pid'], p2['pid']]) mydb.close_db() pass
def add_pattern(ps): mydb.open() pid = mydb.insert_db('INSERT INTO pattern (creator) VALUES (?)',['spikewang']) print pid nid = mydb.insert_db('INSERT INTO node (pid,node_index,active) VALUES (?,1,1)', [pid]) for p in ps: mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)',[pid,p]) mydb.close_db() return (pid, nid)
def change_security_question(index, q, a, username): if len(a) == 0: return -1 mydb.open() query = 'update account set sq_%d= ?,as_%d=? WHERE username = ?' % (index, index) print query mydb.insert_db(query, [q, a, username]) mydb.close_db() return 1
def store_message_in_database(_from, _to, msg): mydb.open() ISOTIMEFORMAT='%Y-%m-%d %X' createdTime=str(time.strftime( ISOTIMEFORMAT, time.localtime() )) result = mydb.insert_db('insert into message(_from,_to,msg,createdAt) values(?,?,?,?)', [_from,_to,msg,createdTime]) mydb.close_db()
def createAccount(list): mydb.open() if len(list[4])==0: return ["First name can't be null", 0] if len(list[5])==0: return ["Last name can't be null",1] result = _valid_login_format(list[0],list[1]) # if error happend return error message if not result == None : return result if not check_username(list[0]): return ["Username has already existed",2] mydb.insert_db("""insert into account(username,pwd,type,status,fname,lname) values(?,?,?,?,?,?)""", [list[0], list[1], list[2], list[3], list[4], list[5]]) mydb.close_db() return None
def add_domain(ds): mydb.open() did = mydb.insert_db('INSERT INTO domain (creator) VALUES (?)',['spikewang']) print did d_pid = mydb.insert_db('INSERT INTO pattern (did, isDomain) VALUES (?,1)', [did]) d_nid = mydb.insert_db('INSERT INTO node (pid, node_index, active) VALUES (?,1,1)', [d_pid]) pid = mydb.insert_db('INSERT INTO pattern (did, isDomain) VALUES (?,0)', [did]) nid = mydb.insert_db('INSERT INTO node (pid, node_index, active) VALUES (?,1,1)', [pid]) for d in ds: d_pattern = mydb.query_db('SELECT * FROM pattern WHERE did = ? and isDomain = 1', [d], one = True) mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)', [d_pid, d_pattern['pid']]) mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)', [d_pid, pid]) mydb.close_db() return (did, pid, nid, d_nid)
def add_node(pid,index): mydb.open() results = mydb.query_db('SELECT * FROM node WHERE pid = ? and node_index = ?', [pid, index/2] ) if len(results)!=0: nid = mydb.insert_db('INSERT INTO node (pid,node_index,active) VALUES (?,?,1)',[pid,index]) mydb.close_db() return nid else: return -1
def add_pattern(ps, did): mydb.open() d_pattern = mydb.query_db('SELECT * FROM pattern WHERE did = ? and isDomain = 1', [did], one = True) pid = mydb.insert_db('INSERT INTO pattern (did, isDomain) VALUES (?,0)', [did]) nid = mydb.insert_db('INSERT INTO node (pid,node_index,active) VALUES (?,1,1)', [pid]) mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)', [pid, d_pattern['pid']]) for p in ps: mydb.insert_db('INSERT INTO edge(_from, _to) VALUES (?,?)',[pid,p]) mydb.close_db() return (pid, nid)
def add_node(): mydb.open() nid = mydb.insert_db('INSERT INTO pattern (pid,node_index,active,x,y) VALUES (1,1,1,1,1)') mydb.close_db()
def change_node_status(nid, active): mydb.open() # active =1 means it is active result = mydb.insert_db('update node SET active=? WHERE nid=?', [active,nid]) mydb.close_db()
import mydb, sqlite3 from models import accessControl # mydb.insert_db("""update account SET pwd = ? WHERE username = ? """, # ('wzs','spikewang')) # # print accessControl.verify('spikewang','dsd').name mydb.open() id = 'A0' mydb.insert_db("""UPDATE node SET x=? WHERE id = ? """, (2, id)) mydb.close_db()
def active_all_Nodes(): mydb.open() result = mydb.insert_db('update node SET active=1') mydb.close_db()
def addAccount(list): mydb.open() result = mydb.insert_db("""insert into account(username,pwd,type,status) values(?,?,?,?)""",[list[0],list[1],list[2],list[3]] ) mydb.close_db()
def lockAccount(username): mydb.open() result = mydb.insert_db("""update account set status=0 WHERE username= ?""", (username,)) mydb.close_db() return result
def createSecQues(list,username): mydb.open() mydb.insert_db("""update account set sq_1= ?,as_1=? ,sq_2=?, as_2=?,sq_3=? ,as_3=? WHERE username= ?""", [list[0], list[1], list[2], list[3], list[4], list[5], username]) mydb.close_db()
def send_message(_from, _to, msg): mydb.open() result = mydb.insert_db('insert into message(from_id,to_id,msg) values(?,?,?)', [_from,_to,msg]) mydb.close_db()