adding stuff that was missing

Author: Dan Finn

__pycache__/users.cpython-35.pyc

@@ -0,0 +1,7 @@
+import sqlite3
+
+db = sqlite3.connect('guestbook.db')
+db.execute('create table users (id INTEGER PRIMARY KEY autoincrement, username text, password text, date_joined int)')
+db.execute('create table posts (id INTEGER PRIMARY KEY autoincrement, poster_id int, title text, body text, time_posted int)')
+db.commit()
+db.close()

create_database.py

@@ -0,0 +1,31 @@
+import users
+
+def display_menu():
+    print('1: Login')
+    print('2: Register')
+    if users.is_logged_in == True:
+        print('3: Post Comment')
+    print('9: Exit')
+    choice = input('Enter the number of where you want to go: ')
+    check_choice(choice)
+    
+def check_choice(choice):
+    if choice.isdigit():
+        choice = int(choice)
+        if choice == 1:
+            users.login()
+            display_menu()
+        elif choice == 2:
+            users.register()
+            display_menu()
+        elif choice == 3:
+            pass
+        elif choice == 9:
+            pass
+        else:
+            print('Please choose a correct option')
+            display_menu()
+    else:
+        print('Please choose a correct option')
+        display_menu()
+display_menu()

guestbook.db

@@ -0,0 +1,50 @@
+import sqlite3
+db = sqlite3.connect('guestbook.db')
+is_logged_in = False
+user_id = 0 # 0 means not set / not logged in.
+
+def register():
+    username = str(input('Please enter a username: '))
+    if username_available(username) == True :
+        import hashlib, time
+        password = input('Please enter a password: ').encode('utf_8')
+        encrypted = hashlib.sha256()
+        encrypted.update(password)
+        newpass = encrypted.hexdigest()
+        
+        currenttime = int(time.time())
+        db.execute('insert into users (username, password, date_joined) values (?,?,?)',(username,newpass, currenttime))
+        db.commit()
+        print('You have been added to the user database')
+        return True
+    else:
+        print('Unfortunately, that username is already taken. Please try again')
+        register()
+def login():
+    username = input('Please enter your username: ')
+    if username_available(username) == False: #if its taken, then it must exist.
+        import hashlib
+        password = input('Please enter your password: ').encode('utf_8')
+        encrypted = hashlib.sha256()
+        encrypted.update(password)
+        newpass = encrypted.hexdigest()
+        
+        db.row_factory = sqlite3.Row
+        row = db.execute('select id from users where username = ? and password = ?', (username, newpass))
+        if row is None:
+            print('Sorry, that password doesn\' match the username')
+        else:
+            user_id = row.fetchone()['id']
+            is_logged_in = True
+    else:
+        print('That user doesn\'t exist. Register it?')
+def logout():
+    pass
+def username_available(username):
+    db.row_factory = sqlite3.Row
+    row = db.execute('select id from users WHERE username = ?',(username,))
+    
+    if row.fetchone() is None:
+        return True
+    else:
+        return False