12
12
import ea_tasks
13
13
import pandas as pd
14
14
import numpy as np
15
+ import backup
16
+
15
17
16
18
def create_token_table (delete = False ):
17
19
with open ('config/mysqlconfig.yaml' , 'r' ) as cfile :
@@ -26,26 +28,27 @@ def create_token_table(delete=False):
26
28
con .select_db ('des' )
27
29
cur = con .cursor ()
28
30
if delete :
29
- cur .execute ("DROP TABLE IF EXISTS Tokens " )
31
+ cur .execute ("DROP TABLE IF EXISTS apiTokens " )
30
32
cur .execute ("""
31
- CREATE TABLE IF NOT EXISTS Tokens (
33
+ CREATE TABLE IF NOT EXISTS apiTokens (
32
34
token varchar(50),
33
35
time datetime,
36
+ user varchar(100),
34
37
)""" )
35
38
con .commit ()
36
39
con .close ()
37
40
38
41
39
- def create_token ():
42
+ def create_token (user = 'test' ):
40
43
token = hashlib .sha1 (os .urandom (64 )).hexdigest ()
41
44
now = datetime .datetime .now ()
42
45
with open ('config/mysqlconfig.yaml' , 'r' ) as cfile :
43
46
conf = yaml .load (cfile )['mysql' ]
44
47
con = mydb .connect (** conf )
45
- tup = tuple ([token , now .strftime ('%Y-%m-%d %H:%M:%S' )])
48
+ tup = tuple ([token , now .strftime ('%Y-%m-%d %H:%M:%S' ), user ])
46
49
with con :
47
50
cur = con .cursor ()
48
- cur .execute ("INSERT INTO Tokens VALUES {0}" .format (tup ))
51
+ cur .execute ("INSERT INTO apiTokens VALUES {0}" .format (tup ))
49
52
con .close ()
50
53
51
54
@@ -96,7 +99,7 @@ def post(self):
96
99
con = mydb .connect (** conf )
97
100
try :
98
101
cur = con .cursor ()
99
- cur .execute ("SELECT * from Tokens where token = '{0}'" .format (token ))
102
+ cur .execute ("SELECT * from apiTokens where token = '{0}'" .format (token ))
100
103
cc = cur .fetchone ()
101
104
now = datetime .datetime .now ()
102
105
dt = (now - cc [1 ]).total_seconds ()
@@ -194,7 +197,7 @@ def post(self):
194
197
con = mydb .connect (** conf )
195
198
try :
196
199
cur = con .cursor ()
197
- cur .execute ("SELECT * from Tokens where token = '{0}'" .format (token ))
200
+ cur .execute ("SELECT * from apiTokens where token = '{0}'" .format (token ))
198
201
cc = cur .fetchone ()
199
202
now = datetime .datetime .now ()
200
203
dt = (now - cc [1 ]).total_seconds ()
@@ -264,7 +267,7 @@ def post(self):
264
267
con = mydb .connect (** conf )
265
268
try :
266
269
cur = con .cursor ()
267
- cur .execute ("SELECT * from Tokens where token = '{0}'" .format (token ))
270
+ cur .execute ("SELECT * from apiTokens where token = '{0}'" .format (token ))
268
271
cc = cur .fetchone ()
269
272
now = datetime .datetime .now ()
270
273
dt = (now - cc [1 ]).total_seconds ()
0 commit comments