-
Notifications
You must be signed in to change notification settings - Fork 0
/
Auth.py
97 lines (83 loc) · 2.69 KB
/
Auth.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# -*- coding: utf-8 -*-
import os
import json
import Users
from Users import connect
def create_table():
conn = connect()
cursor = conn.cursor()
cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
tables = cursor.fetchall()
if not ('authorization',) in tables:
cursor.execute('CREATE TABLE authorization ('
'id INTEGER PRIMARY KEY AUTOINCREMENT, '
'filename TEXT NOT NULL, '
'username TEXT NOT NULL,'
'auth INTEGER);')
conn.commit()
cursor.close()
conn.close()
return
def have_edit_auth(filename, username):
conn = connect()
cursor = conn.cursor()
s = "SELECT auth FROM authorization WHERE filename='" + filename + "' AND username='" + username + "';"
cursor.execute(s)
ret = cursor.fetchall()
cursor.close()
conn.close()
if len(ret) == 0:
return 0
return ret[0][0] >= 1
def have_manage_auth(filename, username):
conn = connect()
cursor = conn.cursor()
s = "SELECT auth FROM authorization WHERE filename='" + filename + "' AND username='" + username + "';"
cursor.execute(s)
ret = cursor.fetchall()
cursor.close()
conn.close()
if len(ret) == 0:
return 0
return ret[0][0] >= 2
def get_edit_list(username):
conn = connect()
cursor = conn.cursor()
s = "SELECT filename FROM authorization WHERE username='" + username + "';"
cursor.execute(s)
ret = cursor.fetchall()
cursor.close()
conn.close()
return ret
def change(filename, username, auth, myname):
if auth == -1:
for user in Users.all_users():
if user[0] != myname:
change(filename, user[0], 1, myname)
return
if auth == -2:
for user in Users.all_users():
if user[0] != myname:
change(filename, user[0], 0, myname)
return
conn = connect()
cursor = conn.cursor()
s = "SELECT auth FROM authorization WHERE filename='" + filename + "' AND username='" + username + "';"
# print s
cursor.execute(s)
ret = cursor.fetchall()
if len(ret) == 0:
s = "INSERT INTO authorization (filename, username, auth) " + \
"values ('" + filename + "', '" + username + "', " + str(auth) + ");"
cursor.execute(s)
conn.commit()
else:
s = "UPDATE authorization SET auth=" + str(auth) + \
" WHERE filename='" + filename + "' AND username='" + username + "';"
# print s
cursor.execute(s)
conn.commit()
# create_table()
# print Users.all_users()
# change('a.txt', 'zwt', -2, 'zwt')
# change('a.txt', 'zwt', 2, 'zwt')