update init, add logo

install.py

@@ -1,12 +1,26 @@
-#!/usr/bin/python
+#!/srv/pandora/bin/python3
 
 import os
+import re
 from os.path import join, abspath, basename, dirname
 
-name = 'leftovers'
+import ox
+
+
+overwrite = (
+    #('home', 'indiancinema'),
+    #('infoView', 'indiancinema'),
+)
+
 base = abspath(dirname(__file__))
 os.chdir(base)
 
+name = os.path.basename(base)
+config = join(base, 'config.jsonc')
+if os.path.exists(config):
+    with open(config) as fd:
+        name = ox.jsonc.load(fd)['site']['id']
+
 for root, folders, files in os.walk(join(base, 'static')):
     for f in files:
         src = join(root, f)
@@ -14,22 +28,35 @@ for root, folders, files in os.walk(join(base, 'static')):
         rel_src = os.path.relpath(src, dirname(target))
         if os.path.exists(target):
             os.unlink(target)
+        target_folder = os.path.dirname(target)
+        if not os.path.exists(target_folder):
+            os.makedirs(target_folder)
         os.symlink(rel_src, target)
 
-'''
-overwrite = (
-    ('home', 'indiancinema'),
-    ('infoView', 'indiancinema'),
-)
+for img in ('logo.png', 'icon.png'):
+    f = join('/srv/pandora/static/png', img)
+    custom = join('/srv/pandora/static/png', img.replace('.png', '.%s.png' % name))
+    if os.path.exists(custom):
+        if not os.path.islink(f):
+            os.unlink(f)
+        else:
+            path = os.path.realpath(f)
+            if name not in path:
+                os.unlink(f)
+        if not os.path.exists(f):
+            os.symlink(os.path.basename(custom), f)
 
-os.chdir('/srv/pandora/static/js')
-for filename, sitename in overwrite:
-    src = '%s.%s.js' % (filename, sitename)
-    target = '%s.%s.js' % (filename, name)
-    if os.path.exists(target):
-        os.unlink(target)
-    os.symlink(src, target)
-'''
+if overwrite:
+    os.chdir('/srv/pandora/static/js')
+    for filename, sitename in overwrite:
+        src = '%s.%s.js' % (filename, sitename)
+        target = '%s.%s.js' % (filename, name)
+        if os.path.exists(target):
+            os.unlink(target)
+        target_folder = os.path.dirname(target)
+        if not os.path.exists(target_folder):
+            os.makedirs(target_folder)
+        os.symlink(src, target)
 
 os.chdir(base)
 src = join(base, 'config.jsonc')
@@ -46,15 +73,40 @@ os.symlink(basename(target), t)
 for root, folders, files in os.walk(join(base, 'scripts')):
     for f in files:
         src = join(root, f)
-        if '__pycache__' in src or src.endswith('.pyc'):
-            continue
         target = src.replace(base, '/srv/pandora')
         rel_src = os.path.relpath(src, dirname(target))
         if os.path.exists(target):
             os.unlink(target)
         os.symlink(rel_src, target)
+        if f == 'poster.%s.py' % name:
+            t = os.path.join(dirname(target), 'poster.py')
+            if os.path.exists(t):
+                os.unlink(t)
+            os.symlink(f, os.path.join(dirname(target), t))
 
-#todo
-#custom python module etc
-#local_settings.py?
 
+if os.path.exists('__init__.py'):
+    # make module available to pandora
+    target = os.path.join('/srv/pandora/pandora/', name)
+    rel_src = os.path.relpath(base, dirname(target))
+    if os.path.exists(target):
+        os.unlink(target)
+    os.symlink(rel_src, target)
+
+    # include module in local settings
+    local_settings_py = '/srv/pandora/pandora/local_settings.py'
+    with open(local_settings_py) as fd:
+        local_settings_changed = False
+        local_settings = fd.read()
+        if 'LOCAL_APPS' not in local_settings:
+            local_settings += '\nLOCAL_APPS = ["%s"]\n' % name
+            local_settings_changed = True
+        else:
+            apps = re.compile('(LOCAL_APPS.*?)\]', re.DOTALL).findall(local_settings)[0]
+            if name not in apps:
+                new_apps = apps.strip() + ',\n"%s"\n' % name
+                local_settings = local_settings.replace(apps, new_apps)
+                local_settings_changed = True
+    if local_settings_changed:
+        with open(local_settings_py, 'w') as fd:
+            fd.write(local_settings)