Maintain user admin with the admin server password

src/pyscripts/xsbs/plugins.py

@@ -30,6 +30,7 @@ def main():
 	pm.loadPlugins()
 	setup_all()
 	create_all()
+	xsbs.server.setupAdminUser()
 
 main()
 

src/pyscripts/xsbs/server.py

@@ -4,6 +4,11 @@
 from xsbs.ui import notice
 from xsbs.colors import colordict
 from xsbs.settings import PluginConfig
+from xsbs.users import User
+from xsbs.users.privilege import UserPrivilege
+
+from sqlalchemy.orm.exc import NoResultFound
+from elixir import session
 
 import string
 import sbserver
@@ -20,6 +25,23 @@
 	'private',
 ]
 
+def setupAdminUser():
+	try:
+		adminUser = User.query.filter(User.email=="admin").one()
+	except NoResultFound:
+		adminUser = User("admin", sbserver.adminPassword())
+		session.commit()
+		adminUser = User.query.filter(User.email=="admin").one()
+	else:
+		if adminUser.password != sbserver.adminPassword():
+			adminUser.password = sbserver.adminPassword()
+			adminUser.update()
+	try:
+		adminPriv = UserPrivilege.query.filter(UserPrivilege.user_id==adminUser.id).one()
+	except NoResultFound:
+		adminPriv = UserPrivilege(2, adminUser.id)
+		session.commit()
+
 def isPaused():
 	'''Is the game currently paused'''
 	return sbserver.isPaused()
@@ -121,4 +143,4 @@ def csevalfile(file):
 		cseval(f.read())
 	except: pass
 	finally:
-		f.close()
+		f.close()