anatolykopyl/DINS-test-assignment
1
0
Fork 0
mirror of https://github.com/anatolykopyl/DINS-test-assignment.git synced 2026-03-26 12:55:05 +00:00
Code Issues Projects Releases Wiki Activity

Added a check for MySQL table existance

Browse Source
This commit is contained in:
Anatoly Kopyl
2017-11-16 01:22:14 +03:00
parent 25a437e283
commit 474ca0b713
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@@ -16,7 +16,7 @@ with open('raw_data2.csv', 'rt', encoding="UTF-8") as csvfile:
name = row[1] + "*" + row[2] # combining api_name and http_method into a pair name name = row[1] + "*" + row[2] # combining api_name and http_method into a pair name
if 'start_time' not in locals(): if 'start_time2' not in locals():
start_time = timestamp start_time = timestamp
start_time = int(min(start_time, timestamp)) # earliest request start_time = int(min(start_time, timestamp)) # earliest request
if 'end_time' not in locals(): if 'end_time' not in locals():
@@ -45,6 +45,9 @@ is_anomaly(result)
try: try:
connection = MySQLdb.connect(host="127.0.0.1", user="user1", passwd="testserver", db="mydb") connection = MySQLdb.connect(host="127.0.0.1", user="user1", passwd="testserver", db="mydb")
cursor = connection.cursor() cursor = connection.cursor()
cursor.execute("SHOW TABLES LIKE 'result'")
if not cursor.fetchone(): # check if table already exists
cursor.execute("CREATE TABLE result ( timeframe_start TIMESTAMP, api_name TEXT, http_method TEXT, \ cursor.execute("CREATE TABLE result ( timeframe_start TIMESTAMP, api_name TEXT, http_method TEXT, \
count_http_code_5xx INT, is_anomaly TINYINT(1) )") count_http_code_5xx INT, is_anomaly TINYINT(1) )")