Merge branch 'custom_settings' of DebenOldert/Music-Downloader into master

helper/console.py

@@ -7,7 +7,7 @@ OUTPUT = 0
 DBG_ERROR = 1
 DBG_INFO = 2
 
-LOCKED = False
+LOCKED = True
 
 CACHE = []
 
@@ -16,10 +16,10 @@ color.init(True)
 
 def output(text='', level=OUTPUT, end='\n'):
     if LOCKED:
-        CACHE.append((text, level, end))
+        #CACHE.append((text, level, end))
         return
 
-    if level <= Settings.Debuglvl:
+    if level <= Settings.Debuglvl.get():
 
         if level == 1:
             text = color.Fore.RED + '[ERROR] ' + text

helper/settings.py

@@ -1,12 +1,57 @@
+import os
+import configparser
+
+
+CONFIG = configparser.ConfigParser()
+if os.path.isfile('settings.ini'):
+    CONFIG.read('settings.ini')
+
+
+class SettingEntry:
+    def __init__(self, name, env_name=None, default=None, namespace='DEFAULT', type=str):
+        self.name = name
+        self.environment = env_name
+        self.default = default
+        self.value = default
+        self.namespace = namespace
+        self.read_from_env = False
+        self.type = type
+
+        self.read_config()
+
+        if os.getenv(self.environment) is not None:
+            self.read_from_env = True
+            self.set(os.getenv(self.environment))
+
+    def __str__(self):
+        return '{} => {}'.format(self.name, self.value)
+
+    def __eq__(self, other):
+        return self.name == other
+
+    def read_config(self):
+        if self.namespace in CONFIG and self.name in CONFIG[self.namespace] and not self.read_from_env:
+            self.set(CONFIG[self.namespace][self.name])
+
+    def set(self, value):
+        self.value = self.type(value)
+
+    def get(self):
+        return self.type(self.value)
+
+
 class Settings:
     # Directory to permanently save the file (ENV: MD_SAVEDIR)
-    SaveDir = None
+    SaveDir = SettingEntry('saveDir', 'MD_SAVEDIR', '~/Downloads', 'music-downloader', os.path.expanduser)
 
     # Directory to temporary download the file (ENV: MD_TMP)
-    tmpDir = '/tmp'
+    tmpDir = SettingEntry('tmpDir', 'MD_TMP', '~/tmp', 'music-downloader', os.path.expanduser)
 
     # Minimal debug level (ENV: MD_LOGGING)
-    Debuglvl = None
+    Debuglvl = SettingEntry('debuglvl', 'MD_LOGGING', 0, 'music-downloader', int)
 
     # Minimal bitrate to auto download the file (ENV: MD_QUALITY)
-    MinQuality = 300
+    MinQuality = SettingEntry('minQuality', 'MD_QUALITY', 300, 'music-downloader', int)
+
+    #Format for downloaded file ID3 comment
+    CommentFormat = SettingEntry('saveDir', 'MD_SAVEDIR', '~/Downloads', 'music-downloader')

helper/sites/download.py

@@ -36,7 +36,7 @@ def download(item, type='GET', parameters=None, headers=None, cookies=None, stre
         console.output('Assuming it\'s a mp3 file', console.DBG_INFO)
         name = name + '.mp3'
 
-    full_name = os.path.abspath('{0}/{1}'.format(Settings.tmpDir, name))
+    full_name = os.path.abspath('{0}/{1}'.format(Settings.tmpDir.get(), name))
 
     if item.bitrate is Empty and item.bytes is None:
         size = file.headers.get('content-length')
@@ -50,7 +50,7 @@ def download(item, type='GET', parameters=None, headers=None, cookies=None, stre
 
                 console.output('Bitrate: {0}kbps'.format(int(item.bitrate)))
 
-            if item.duration_seconds is not None and int(item.bitrate) < Settings.MinQuality:
+            if item.duration_seconds is not None and int(item.bitrate) < Settings.MinQuality.get():
                 ans = console.ask_input('Continue downloading? [y/n]')
 
                 if ans != 'y':
@@ -87,7 +87,7 @@ def savefileprogress(name, full_name, file, item):
 
         name = urllib.parse.unquote('{x.artist} - {x.title}'.format(x=tags)) + '.mp3'
 
-    full_save_name = os.path.abspath('{0}/{1}'.format(Settings.SaveDir, name))
+    full_save_name = os.path.abspath('{0}/{1}'.format(Settings.SaveDir.get(), name))
     os.rename(full_name, full_save_name)
     console.output('Moved to {0}'.format(full_save_name), level=console.DBG_INFO)
 

main.py

@@ -1,32 +1,18 @@
 import sites as sites
+from helper.settings import Settings
 from helper import console as console
 import os.path
-from helper.settings import Settings
-
-
-def init():
-    dbg_lvl = os.getenv('MD_LOGGING', 'ERROR')
-
-    if dbg_lvl.lower() == 'info':
-        Settings.Debuglvl = console.DBG_INFO
-    else:
-        Settings.Debuglvl = console.DBG_ERROR
-
-    Settings.tmpDir = os.path.expanduser(os.getenv('MD_TMP', '/tmp'))
-
-    os.makedirs(Settings.tmpDir, exist_ok=True)
-
-    directory(os.getenv('MD_SAVEDIR', '~/Downloads'))
-
-    quality(os.getenv('MD_QUALITY', '300'))
 
 
 def main():
-    init()
+    os.makedirs(Settings.tmpDir.get(), exist_ok=True)
+    console.unlock()
+
 
     console.output('Music downloader by Deben Oldert')
     console.output()
-    console.output('Download directory: {0}'.format(Settings.SaveDir))
+
+    console.output('Download directory: {0}'.format(Settings.SaveDir.get()))
     console.output()
     help()
     console.output()
@@ -130,17 +116,17 @@ def search(cache=None):
 def directory(dir=None):
     if dir is None:
         dir = console.ask_input('Enter new save directory')
-    Settings.SaveDir = os.path.expanduser(dir)
-    os.makedirs(Settings.SaveDir, exist_ok=True)
-    console.output('New save directory: {0}'.format(Settings.SaveDir), level=console.DBG_INFO)
+    Settings.SaveDir.set(dir)
+    os.makedirs(Settings.SaveDir.get(), exist_ok=True)
+    console.output('New save directory: {0}'.format(Settings.SaveDir.get()), level=console.DBG_INFO)
     return True
 
 
 def quality(quality=None):
     if quality is None:
         quality = console.ask_input('Enter minimal quality for auto download')
-    Settings.MinQuality = int(quality)
-    console.output('Minimal quality for auto downloading is set to {0}'.format(Settings.MinQuality), level=console.DBG_INFO)
+    Settings.MinQuality.set(quality)
+    console.output('Minimal quality for auto downloading is set to {0}'.format(Settings.MinQuality.get()), level=console.DBG_INFO)
     return True
 
 

settings.ini

@@ -0,0 +1,6 @@
+[music-downloader]
+saveDir=~/Downloads
+tmpDir=/tmp
+debuglvl=0
+minQuality=300
+commentFormat=LABEL:{label};