create required debugproxy organization in init_db

Author: tarnacious

proxyweb/startup/create_users.py

@@ -5,6 +5,14 @@
 from typing import List, Set, Dict, Tuple, Text, Optional, Any
 
 
+def create_debugproxy_organization():
+    organization = Organization.query.filter(Organization.name == name).first()
+    if not organization:
+        organization = Organization(name=name)
+        db.session.add(organization)
+        db.session.commit()
+
+
 def create_admin_command(email, password) -> None:
     admin_role = find_or_create_role('admin', u'Admin')
     system_admin_role = find_or_create_role('system_admin', u'System Admin')

proxyweb/startup/manager.py

@@ -2,7 +2,7 @@
 import database as db
 from proxyweb.startup.create_users import find_or_create_organization, \
     find_or_create_user, create_user_command, create_admin_command, \
-    create_session_command
+    create_session_command, create_debugproxy_organization
 
 from proxyweb.startup.create_users import create_users
 from database.models import Organization, User
@@ -11,21 +11,19 @@
 @manager.command
 def init_db(): # type: ignore
     db.create_all()
-
+    create_debugproxy_organization("debugproxy")
 
 @manager.command
 def drop_db(): # type: ignore
     db.session.commit()
     db.drop_all()
 
-
 @manager.command
 def clear_db(): # type: ignore
     db.session.query(Organization).delete()
     db.session.query(User).delete()
     db.session.commit()
 
-
 @manager.command
 def create_admin(email, password):
     create_admin_command(email, password)