From 5a735e46cae340d0f212d5f738ed345ffb9bda86 Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@freedombone.net>
Date: Sat, 25 Jul 2020 11:13:24 +0100
Subject: [PATCH] Simplify

---
 theme.py | 86 ++++++++++++++++----------------------------------------
 1 file changed, 24 insertions(+), 62 deletions(-)

diff --git a/theme.py b/theme.py
index cfe30544..9947c12f 100644
--- a/theme.py
+++ b/theme.py
@@ -592,14 +592,6 @@ def setThemeImages(baseDir: str, name: str) -> None:
     themeNameLower = name.lower()
 
     if themeNameLower == 'default':
-        loginBackgroundImageFilename = \
-            baseDir + '/img/login-background.png'
-        sharesBackgroundImageFilename = \
-            baseDir + '/img/shares-background.png'
-        deleteBackgroundImageFilename = \
-            baseDir + '/img/delete-background.png'
-        followBackgroundImageFilename = \
-            baseDir + '/img/follow-background.png'
         profileImageFilename = \
             baseDir + '/img/image.png'
         bannerFilename = \
@@ -607,14 +599,6 @@ def setThemeImages(baseDir: str, name: str) -> None:
         searchBannerFilename = \
             baseDir + '/img/search_banner.png'
     else:
-        loginBackgroundImageFilename = \
-            baseDir + '/img/login_background_' + themeNameLower + '.png'
-        sharesBackgroundImageFilename = \
-            baseDir + '/img/shares_background_' + themeNameLower + '.png'
-        deleteBackgroundImageFilename = \
-            baseDir + '/img/delete_background_' + themeNameLower + '.png'
-        followBackgroundImageFilename = \
-            baseDir + '/img/follow_background_' + themeNameLower + '.png'
         profileImageFilename = \
             baseDir + '/img/image_' + themeNameLower + '.png'
         bannerFilename = \
@@ -632,53 +616,31 @@ def setThemeImages(baseDir: str, name: str) -> None:
             accountDir = \
                 os.path.join(baseDir + '/accounts', acct)
 
-            if os.path.isfile(loginBackgroundImageFilename):
-                try:
-                    copyfile(loginBackgroundImageFilename,
-                             accountDir + '/login-background.png')
-                except BaseException:
-                    pass
-            elif os.path.isfile(accountDir + '/login-background.png'):
-                try:
-                    os.remove(accountDir + '/login-background.png')
-                except BaseException:
-                    pass
+            backgroundNames = ('login', 'shares', 'delete', 'follow',
+                               'options', 'block', 'search', 'calendar')
+            for backgroundType in backgroundNames:
+                if themeNameLower == 'default':
+                    backgroundImageFilename = \
+                        baseDir + '/img/' + backgroundType + '-background.png'
+                else:
+                    backgroundImageFilename = \
+                        baseDir + '/img/' + backgroundType + '_background_' + \
+                        themeNameLower + '.png'
 
-            if os.path.isfile(sharesBackgroundImageFilename):
-                try:
-                    copyfile(sharesBackgroundImageFilename,
-                             accountDir + '/shares-background.png')
-                except BaseException:
-                    pass
-            elif os.path.isfile(accountDir + '/shares-background.png'):
-                try:
-                    os.remove(accountDir + '/shares-background.png')
-                except BaseException:
-                    pass
-
-            if os.path.isfile(deleteBackgroundImageFilename):
-                try:
-                    copyfile(deleteBackgroundImageFilename,
-                             accountDir + '/delete-background.png')
-                except BaseException:
-                    pass
-            elif os.path.isfile(accountDir + '/delete-background.png'):
-                try:
-                    os.remove(accountDir + '/delete-background.png')
-                except BaseException:
-                    pass
-
-            if os.path.isfile(followBackgroundImageFilename):
-                try:
-                    copyfile(followBackgroundImageFilename,
-                             accountDir + '/follow-background.png')
-                except BaseException:
-                    pass
-            elif os.path.isfile(accountDir + '/follow-background.png'):
-                try:
-                    os.remove(accountDir + '/follow-background.png')
-                except BaseException:
-                    pass
+                if os.path.isfile(backgroundImageFilename):
+                    try:
+                        copyfile(backgroundImageFilename,
+                                 accountDir + '/' + backgroundType +
+                                 '-background.png')
+                    except BaseException:
+                        pass
+                elif os.path.isfile(accountDir + '/' + backgroundType +
+                                    '-background.png'):
+                    try:
+                        os.remove(accountDir + '/' + backgroundType +
+                                  '-background.png')
+                    except BaseException:
+                        pass
 
             if os.path.isfile(profileImageFilename) and \
                os.path.isfile(bannerFilename):