Fix/update site script (#98)

* Update site name and domain on script Signed-off-by: Matheus Fernandes <[email protected]> * Add missed function cal Signed-off-by: Matheus Fernandes <[email protected]>
labhackercd · Nov 28, 2017 · ac58f3b · ac58f3b
1 parent 41e2301
commit ac58f3b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/wikilegis/create_admin.py b/wikilegis/create_admin.py
@@ -25,6 +25,26 @@ def create_superuser():
         sys.exit(1)
 
 
+def update_sites():
+    from django.contrib.sites.models import Site
+    import re
+    site = Site.objects.get_current()
+    site_domain = os.environ.get('SITE_DOMAIN', None)
+    site_name = os.environ.get('SITE_NAME', None)
+
+    if None not in [site_domain, site_name]:
+        print('Updating site infos...')
+        site_domain = re.sub('^(http|https)://', '', site_domain)
+
+        site.domain, site.name = site_domain, site_name
+        site.save()
+        print('Done!')
+    else:
+        print('Missing SITE_DOMAIN or SITE_NAME environment variable.')
+        sys.exit(2)
+
+
 if __name__ == '__main__':
     django.setup()
     create_superuser()
+    update_sites()