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()
