Ignore testservers on import

1 files changed, 20 insertions, 7 deletions

diff --git a/import/import.py b/import/import.py
index 68324c5..24a871c 100644
--- a/import/import.py
+++ b/import/import.py
@@ -12,22 +12,34 @@ import datetime
 def default(x, default):
     return default if x is None else x
 
-with open(sys.argv[1], 'r') as f:
-    data = json.load(f)
-
 c = configparser.ConfigParser()
 c.read('config.ini')
 
 filename = os.path.basename(sys.argv[1])
 
-date = re.search('^[0-9]{4}-[0-9]{2}-[0-9]{2}', filename).group(0)
-date = datetime.datetime.fromtimestamp(int(data['days7']['tsTo'])).date()
-
 ip = re.search('_([0-9]{1,3}.[0-9]{1,3}.[0-9]{1,3}.[0-9]{1,3})[_\.]', filename).group(1)
 
 db = pymysql.connect(c.get('db', 'host'), c.get('db', 'username'), c.get('db', 'password'), c.get('db', 'name'), cursorclass=pymysql.cursors.DictCursor)
 cursor = db.cursor()
 
+# Check if ip is already in names table
+cursor.execute('SELECT testserver FROM names WHERE ip = %s', (ip))
+entry = cursor.fetchall()
+if len(entry) > 0: # Already in table
+    if entry[0]['testserver'] == 1:
+        print('This is a Testserver. Ignoring!')
+        db.close()
+        exit(0)
+else: # Add to database
+    cursor.execute('INSERT INTO names(ip, name, testserver) VALUES (%s, %s, 0)', (ip, ip))
+
+# Load data and timestamps
+with open(sys.argv[1], 'r') as f:
+    data = json.load(f)
+
+date = re.search('^[0-9]{4}-[0-9]{2}-[0-9]{2}', filename).group(0)
+date = datetime.datetime.fromtimestamp(int(data['days7']['tsTo'])).date()
+
 # Add report to database
 try:
     cursor.execute('INSERT INTO reports(date,ip,version) VALUES(%s, %s, %s)',(
@@ -37,7 +49,8 @@ try:
         ))
 except pymysql.err.IntegrityError:
     print('Report is already in database!')
-    exit(1)
+    db.close()
+    exit(0)
 cursor.execute('SELECT LAST_INSERT_ID() as id')
 reportId = cursor.fetchone()['id']