forked from dodson618/OpenRAT_GmailDev
-
Notifications
You must be signed in to change notification settings - Fork 0
/
regsim_code_review.py
67 lines (55 loc) · 1.51 KB
/
regsim_code_review.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
#! /usr/bin/python
#IMPORT STUFF
import psycopg2, time, datetime, random
DATABASE = 'test_database'
USER = 'andie'
HOST = 'localhost'
PASSWORD = 'h3lly34h'
def initialize_connection(db, usr, hst, pswd):
#Setup DB Connection
conn = psycopg2.connect(database=db, user=usr, host=hst, password=pswd)
return conn.cursor()
### Function Definitions
#Define DB queries
def write(status, cursor):
if status:
transaction_status = '0'
else:
transaction_status = '1'
raw_sql = "INSERT INTO transactionsimulation (tsn_tstate) VALUES (%(transaction_status));"
cursor.execute(raw_sql)
cursor.commit()
def write_dirty(cursor):
write(False, cursor)
def write_clean(cursor):
write(True, cursor)
#Define rolldice
def rolldice(thresh, cursor):
if random.randrange(100) <= thresh:
write_dirty(cursor)
return 1
else:
write_clean(cursor)
return 0
def main():
cursor = initialize_connection(DATABASE, USER, HOST, PASSWORD)
#main loop
Ti = datetime.datetime.now()
dT = datetime.timedelta(0,10)
Tt = datetime.datetime.now() + datetime.timedelta(0,1200)
#initiate main loop
thresh = 25
while datetime.datetime.now() < Tt:
#main program function
#thresh = 25
rolldice(thresh, cursor)
#if a == 0:
# thresh += 25
#else:
# thresh = 25
T0 = datetime.datetime.now()
while datetime.datetime.now() < T0 + dT:
continue
conn.close()
if __name__ == '__main__':
main()