Merge pull request #32 from d-Rickyy-b/autocreateDB

Autocreatedb

Author: devcarron

database/db_wrapper.py

@@ -11,10 +11,46 @@ class __DBwrapper(object):
 
         def __init__(self):
             database_path = os.path.join(self.dir_path, "users.db")
+
+            if not os.path.exists(database_path):
+                print("File '" + database_path + "' does not exist! Trying to create one.")
+                try:
+                    self.create_database(database_path)
+                except:
+                    print("An error has occured while creating the database!")
+
             self.connection = sqlite3.connect(database_path)
             self.connection.text_factory = lambda x: str(x, 'utf-8', "ignore")
             self.cursor = self.connection.cursor()
 
+        def create_database(self, database_path):
+            # Create database file and add admin and users table to the database
+            open(database_path, 'a').close()
+
+            connection = sqlite3.connect(database_path)
+            connection.text_factory = lambda x: str(x, 'utf-8', "ignore")
+            cursor = connection.cursor()
+
+            cursor.execute("CREATE TABLE 'admins' "
+                           "('userID' INTEGER NOT NULL,"
+                           "'first_name' TEXT,"
+                           "'username' TEXT,"
+                           "PRIMARY KEY('userID'));")
+
+            cursor.execute("CREATE TABLE 'users'"
+                           "('userID' INTEGER NOT NULL,"
+                           "'languageID' TEXT,"
+                           "'first_name' TEXT,"
+                           "'last_name' TEXT,"
+                           "'username' TEXT,"
+                           "'gamesPlayed' INTEGER,"
+                           "'gamesWon' INTEGER,"
+                           "'gamesTie' INTEGER,"
+                           "'lastPlayed' INTEGER,"
+                           "PRIMARY KEY('userID'));")
+            connection.commit()
+            connection.close()
+
         def get_user(self, user_id):
             self.cursor.execute("SELECT * FROM users WHERE userID=?;", [str(user_id)])