Changed the sys.exit, if not tmpstr, the errorhandling, the creation …

…of the Target Directory, and some PEP8 Styling things too..
beedaddy · Dec 25, 2020 · aba237d · aba237d
1 parent 5ba2de2
commit aba237d
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 20 deletions.
diff --git a/radiorec.py b/radiorec.py
@@ -29,11 +29,14 @@
 import urllib3
 import logging
 import time
+from urllib3.exceptions import MaxRetryError
 logging.basicConfig(level=logging.DEBUG)
 
+
 def print_time():
     return time.strftime("%Y-%m-%d %H:%M:%S")
 
+
 def check_duration(value):
     try:
         value = int(value)
@@ -58,7 +61,8 @@ def read_settings(args):
     elif sys.platform == 'win32':
         settings_base_dir = os.getenv('LOCALAPPDATA') + os.sep + 'radiorec'
     elif sys.platform == 'darwin':
-        settings_base_dir = os.getenv('HOME') + os.sep + 'Library' + os.sep + 'Application Support' + os.sep + 'radiorec'
+        settings_base_dir = os.getenv('HOME') + os.sep + 'Library' + os.sep + \
+                            'Application Support' + os.sep + 'radiorec'
     settings_base_dir += os.sep
     config = configparser.ConfigParser()
     try:
@@ -72,7 +76,7 @@ def read_settings(args):
 
 def record_worker(stoprec, streamurl, target_dir, args):
     pool = urllib3.PoolManager()
-    conn = pool.request('GET',streamurl, preload_content=False)
+    conn = pool.request('GET', streamurl, preload_content=False)
     conn.auto_close = False
     if conn.status != 200:
         conn.release_conn()
@@ -85,7 +89,7 @@ def record_worker(stoprec, streamurl, target_dir, args):
     if args.name:
         filename += '_' + args.name
     content_type = conn.getheader('Content-Type')
-    if(content_type == 'audio/mpeg'):
+    if (content_type == 'audio/mpeg'):
         filename += '.mp3'
     elif(content_type == 'application/aacp' or content_type == 'audio/aacp'):
         filename += '.aac'
@@ -106,10 +110,10 @@ def record_worker(stoprec, streamurl, target_dir, args):
         while(not stoprec.is_set() and not conn.closed):
             target.write(conn.read(1024))
 
-
     verboseprint(print_time() + " ... Connection closed = " + str(conn.closed))
     conn.release_conn()
 
+
 def record(args):
     settings = read_settings(args)
     streamurl = ''
@@ -128,27 +132,33 @@ def record(args):
         try:
             remotefile = pool.request('GET', streamurl)
         except MaxRetryError:
-            print('The URL of the station is not found! Check' + args.station + ' in the Settings!')
+            logging.getLogger(__name__).error('The URL of the station is somehow faulty! Check' +
+                                              args.station + ' in the Settings!')
             sys.exit()
         if remotefile.status != 200:
-            print('The URL of the station is somehow faulty! Check' + args.station + ' in the Settings!')
-            sys.exit()
+            logging.getLogger(__name__).error(
+                'The URL of the station is somehow faulty! Check' + args.station + ' in the Settings!')
+            sys.exit(1)
         else:
             for line in remotefile.data.decode().split():
                 if not line.startswith('#') and len(line) > 1 and line.endswith('mp3'):
                     tmpstr = line
                     break
-        if not len(tmpstr) > 1:
-            print('Could not find a mp3 stream')
-            sys.exit()
+        if not tmpstr:
+            logging.getLogger(__name__).error('Could not find a mp3 stream')
+            sys.exit(1)
         else:
             streamurl = tmpstr
 
     verboseprint(print_time() + " ... Stream URL: " + streamurl)
     target_dir = os.path.expandvars(settings['GLOBAL']['target_dir'])
-    if not os.path.isdir(target_dir + os.sep):
-        print('Target directory not found! Check that ' + target_dir + ' is a valid folder!')
-        sys.exit()
+    if not os.path.isdir(target_dir):
+        try:
+            os.mkdir(target_dir)
+        except FileNotFoundError:
+            logging.getLogger(__name__).error('Target directory not found! Check that ' +
+                                              target_dir + ' is a valid folder!')
+            sys.exit(1)
     started_at = time.time()
     should_end_at = started_at + (args.duration * 60)
     remaining = (args.duration * 60)
@@ -159,18 +169,20 @@ def record(args):
         recthread = threading.Thread(target=record_worker, args=(stoprec, streamurl, target_dir, args))
         recthread.setDaemon(True)
         recthread.start()
-        verboseprint(print_time() + " ... Started thread " + str(recthread) + " timeout: " + str(remaining / 60) + " min")
+        verboseprint(print_time() + " ... Started thread " + str(recthread) + " timeout: " +
+                     str(remaining / 60) + " min")
         recthread.join(remaining)
-        verboseprint(print_time() +  " ... Came out of rec thread again")
+        verboseprint(print_time() + " ... Came out of rec thread again")
 
-        if(recthread.is_alive):
+        if recthread.is_alive:
             stoprec.set()
             verboseprint(print_time() + " ... Called stoprec.set()")
         else:
             verboseprint(print_time() + " ... recthread.is_alive = False")
 
         remaining = should_end_at - time.time()
-        verboseprint(print_time() + " ... Remaining: " + str(remaining / 60) + ", Threads: "  + str(threading.activeCount()))
+        verboseprint(print_time() + " ... Remaining: " + str(remaining / 60) +
+                     ", Threads: " + str(threading.activeCount()))
 
 
 def list(args):
@@ -180,7 +192,8 @@ def list(args):
 
 
 def main():
-    parser = argparse.ArgumentParser(description='This program records internet radio streams. It is free software and comes with ABSOLUTELY NO WARRANTY.')
+    parser = argparse.ArgumentParser(description='This program records internet radio streams. '
+                                                 'It is free software and comes with ABSOLUTELY NO WARRANTY.')
     subparsers = parser.add_subparsers(help='sub-command help')
     parser_record = subparsers.add_parser('record', help='Record a station')
     parser_record.add_argument('station', type=str,
@@ -201,7 +214,8 @@ def main():
     parser_record.set_defaults(func=record)
     parser_list = subparsers.add_parser('list', help='List all known stations')
     parser_list.set_defaults(func=list)
-    parser_list.add_argument('-s', '--settings', nargs='?', type=str, help="specify alternative location for settings.ini")
+    parser_list.add_argument('-s', '--settings', nargs='?', type=str,
+                             help="specify alternative location for settings.ini")
 
     if not len(sys.argv) > 1:
         print('Error: No argument specified.\n')
@@ -210,5 +224,6 @@ def main():
     args = parser.parse_args()
     args.func(args)
 
+
 if __name__ == '__main__':
     main()
diff --git a/settings.ini b/settings.ini
@@ -8,7 +8,7 @@ dkultur = http://www.deutschlandradio.de/streaming/dkultur.m3u
 dlf = http://www.deutschlandradio.de/streaming/dlf.m3u
 dwissen = http://dradio_mp3_dwissen_m.akacast.akamaistream.net/7/728/142684/v1/gnl.akacast.akamaistream.net/dradio_mp3_dwissen_m
 erfplus = http://c14000-l.i.core.cdn.streamfarm.net/14000cina/live/3212erf_96/live_de_96.mp3
-mdrklassik = http://avw.mdr.de/livestreams/mdr_klassik_live_128.m3u
+mdrklassik = http://avw.mdr.de/streams/284350-0_mp3_high.m3u
 radioeins = http://www.radioeins.de/live.m3u
 swr2 = http://mp3-live.swr.de/swr2_m.m3u
 wdr3 = http://wdr-wdr3-live.icecast.wdr.de/wdr/wdr3/live/mp3/256/stream.mp3