Update init script.

Author: jawsper

README.md

@@ -14,7 +14,7 @@ To install:
   * Run `virtualenv -p python3 <location>`.
   * Activate the virtualenv (see virtualenv docs for how to activate on your shell).
 * From the source directory, run `pip install ./barsystem`
-* Run `barsystem-installer`
+* Run `barsystem-installer init`
 * For testing, run barsystem with `django-admin runserver --settings barsystem.local_settings`
 * For production, a standard nginx + uwsgi stack is recommended.
 

barsystem/src/barsystem/install.py

@@ -2,6 +2,9 @@
 import os
 import sys
 
+from django.core.management import execute_from_command_line
+
+
 class BarsystemInstaller:
     def main(self, argv):
         # curses.init_pair(1, curses.COLOR_RED, curses.COLOR_WHITE)
@@ -14,21 +17,93 @@ def main(self, argv):
         #     if key == 'q':
         #         break
 
+        import django
+        django.setup()
+
         self.command = argv[0] if len(argv) else ''
         self.argv = argv[1:] if len(argv) > 1 else []
 
         cmd_name = 'cmd_' + self.command
         if hasattr(self, cmd_name):
             getattr(self, cmd_name)()
         else:
-            print('usage()')
+            self.cmd_help()
+
+    def _command_list(self):
+        return [cmd[len('cmd_'):] for cmd in dir(self) if cmd.startswith('cmd_')]
+
+    def cmd_help(self):
+        print('Possible commands:')
+        for cmd in self._command_list():
+            print(' *', cmd)
 
     def cmd_init(self):
-        print('init!')
+        self.cmd_generate_secret_key()
+        self.cmd_migrate()
+        self.cmd_createsuperuser()
+        self.cmd_create_initial_db_entries()
+
+    def cmd_update(self):
+        print('[Updating barsystem]')
+        self.cmd_migrate()
+
+    def cmd_migrate(self):
+        print('[Running migrations]')
+        execute_from_command_line([
+            sys.argv[0],
+            'migrate'
+        ])
+
+    def cmd_createsuperuser(self):
+        print('[Creating superuser]')
+
+        from django.contrib.auth import get_user_model
+        for user in get_user_model().objects.all():
+            if user.is_superuser:
+                print('- There is already a superuser!')
+                return
+
+        execute_from_command_line([
+            sys.argv[0],
+            'createsuperuser'
+        ])
+        print('- Superuser created.')
 
     def cmd_generate_secret_key(self):
-        settings_module = os.getenv('DJANGO_SETTINGS_MODULE')
-        print('gen sec key!', settings_module)
+        print('[Generating secret key]')
+        from django.conf import settings
+        from django.core.management.utils import get_random_secret_key
+
+        if os.path.exists(settings.SECRET_KEY_FILE):
+            print('- Secret file already exists!')
+            return
+
+        os.makedirs(settings.CONFIG_DIR, exist_ok=True)
+
+        with open(settings.SECRET_KEY_FILE, 'w') as f:
+            f.write(get_random_secret_key())
+        print('- Secret key generated.')
+
+    def cmd_create_initial_db_entries(self):
+        print('[Creating necesarry database entries]')
+        from decimal import Decimal
+        from barsystem.models import Product
+        try:
+            Product.objects.get(special_id='cash_deposit')
+            print('- Cash deposit already exists!')
+        except Product.DoesNotExist:
+            cash_deposit = Product(
+                name='Cash deposit',
+                member_price=Decimal(-1),
+                standard_price=Decimal(-1),
+                active=True,
+                special=True,
+                special_id='cash_deposit',
+                quantity_type='enter_numeric',
+            )
+            cash_deposit.save()
+            print('- Cash deposit created.')
+
 
 
 # Main entrypoint for barsystem-install executable