Move model to cleanup

create_new_key.py

@@ -1,7 +1,5 @@
-from admin_console import cleanup
-from model import generate_keys, unused_keys
+from model import generate_keys, cleanup
 
 if __name__ == '__main__':
     generate_keys(count=1)
-    print(unused_keys())
     cleanup()

create_new_stock.py

@@ -2,7 +2,7 @@ from datetime import timedelta
 from time import strptime
 
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     timeout = float(input('How long will the initial selling be available? (minutes):'))

delete_ownable.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     ownable_id = model.ownable_id_by_name(input('Which ownable?'))

delete_user.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     user_id = model.get_user_id_by_name(input('Which user?'))

drop_old_sessions.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     model.drop_old_sessions()

hash_all_users_passwords.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     if input('Are you sure you want to hash all users passwords? (type in "yes" or something else):') == 'yes':

model.py

@@ -1094,4 +1094,10 @@ def reset_bank():
     cursor.execute('''
         DELETE FROM ownership 
         WHERE user_id = ?
-        ''', (bank_id(),))
+        ''', (bank_id(),))
+
+
+def cleanup():
+    if connection is not None:
+        connection.commit()
+        connection.close()

publish_news_item.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     print(model.new_news(input('Message of the news:')))

reset_bank.py

@@ -1,5 +1,5 @@
 import model
-from admin_console import cleanup
+from model import cleanup
 
 if __name__ == '__main__':
     if input('Are you sure you want to remove all the banks ownerships? (type in "yes" or something else):') == 'yes':