Merge pull request #16 from gjelsas/master

Trying to fix the .m3u decoding problem

README

@@ -8,6 +8,8 @@ Installation
   depending on which platform you are using this program, e.g.:
   * Linux: $HOME/.config/radiorec/settings.ini or
   * Windows: %LOCALAPPDATA%/radiorec/settings.ini
+  or use the commandline option '-s' to specify the location of the
+  settings.ini file.
 - Adjust the settings to your needs. You can happily add more radio stations
   to the STATIONS section.
   !!! Check at least the the target_dir !!!
@@ -17,7 +19,7 @@ Usage
 =====
 Open a shell and go to the directory where radiorec.py is located.
 General usage:
-* Windows: py radirec.py […]
+* Windows: py radiorec.py […]
 * Linux: python3 radiorec.py […] OR JUST ./radiorec.py […]
 
 What you want to do first is getting some help about how to use the scipt:
@@ -37,6 +39,8 @@ Thus the command line is:
 
 You can get a list of all known radio stations with:
 ./radiorec.py list
+You can edit/add the radio stations in the STATIONS section of the settings
+file.
 
 Scheduling the recording task
 =============================
@@ -52,8 +56,8 @@ how to schedule your recording tasks.
 Known problems
 ==============
 The Windows command line (cmd and powershell) still has problems with UTF-8.
-Using the --verbose option might cause the script to crash with an 
-UnicodeEncodeError. If you want to avoid the crash, you have to do both, 
+Using the --verbose option might cause the script to crash with an
+UnicodeEncodeError. If you want to avoid the crash, you have to do both,
 change the command line codepage and the font. For example, doing a
 "chcp 65001" and changing the font to "Lucidia Console" should help.
 

radiorec.py

@@ -26,25 +26,43 @@ import os
 import stat
 import sys
 import threading
-import urllib.request
+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)
     except ValueError:
-        raise argparse.ArgumentTypeError('Duration must be a positive integer.')
+        raise argparse.ArgumentTypeError(
+            'Duration in minutes must be a positive integer.')
 
     if value < 1:
-        raise argparse.ArgumentTypeError('Duration must be a positive integer.')
+        raise argparse.ArgumentTypeError(
+            'Duration in minutes must be a positive integer.')
     else:
         return value
 
-def read_settings():
+
+def read_settings(args):
     settings_base_dir = ''
-    if sys.platform == 'linux':
-        settings_base_dir = os.getenv('HOME') + os.sep + '.config' + os.sep + 'radiorec'
+    if args.settings:
+        settings_base_dir = args.settings
+    elif sys.platform.startswith('linux'):
+        settings_base_dir = os.getenv(
+            'HOME') + os.sep + '.config' + os.sep + 'radiorec'
     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.sep
     config = configparser.ConfigParser()
     try:
@@ -55,15 +73,26 @@ def read_settings():
         sys.exit()
     return dict(config.items())
 
+
 def record_worker(stoprec, streamurl, target_dir, args):
-    conn = urllib.request.urlopen(streamurl)
+    pool = urllib3.PoolManager()
+    conn = pool.request('GET', streamurl, preload_content=False)
+    conn.auto_close = False
+    if conn.status != 200:
+        conn.release_conn()
+        time.sleep(10)
+        verboseprint(print_time() + " ... Waited to return for retry bcof status " + str(conn.status))
+        return
+
     cur_dt_string = datetime.datetime.now().strftime('%Y-%m-%dT%H_%M_%S')
     filename = target_dir + os.sep + cur_dt_string + "_" + args.station
     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'
     elif(content_type == 'application/ogg' or content_type == 'audio/ogg'):
         filename += '.ogg'
     elif(content_type == 'audio/x-mpegurl'):
@@ -71,20 +100,24 @@ def record_worker(stoprec, streamurl, target_dir, args):
         sys.exit()
     else:
         print('Unknown content type "' + content_type + '". Assuming mp3.')
-        filename += 'mp3'
+        filename += '.mp3'
 
-    with open(filename, "wb") as target:
+    verboseprint(print_time() + " ... Writing to: " + filename + ", Content-Type: " + conn.getheader('Content-Type'))
+    with open(filename, 'wb') as target:
         if args.public:
             verboseprint('Apply public write permissions (Linux only)')
-            os.chmod(filename, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP |
-                     stat.S_IROTH | stat.S_IWOTH)
-        verboseprint('Recording ' + args.station + '...')
+            os.chmod(filename, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP | stat.S_IROTH | stat.S_IWOTH)
         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()
+    settings = read_settings(args)
     streamurl = ''
+    tmpstr = ''
     global verboseprint
     verboseprint = print if args.verbose else lambda *a, **k: None
 
@@ -95,48 +128,102 @@ def record(args):
         sys.exit()
     if streamurl.endswith('.m3u'):
         verboseprint('Seems to be an M3U playlist. Trying to parse...')
-        with urllib.request.urlopen(streamurl) as remotefile:
-            for line in remotefile:
-                if not line.decode('utf-8').startswith('#'):
-                    tmpstr = line.decode('utf-8')
+        pool = urllib3.PoolManager()
+        try:
+            remotefile = pool.request('GET', streamurl)
+        except MaxRetryError:
+            logging.getLogger(__name__).error('The URL of the station is somehow faulty! Check' +
+                                              args.station + ' in the Settings!')
+            sys.exit()
+        if remotefile.status != 200:
+            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
-        streamurl = tmpstr
-    verboseprint('stream url: ' + streamurl)
+        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'])
-    stoprec = threading.Event()
+    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)
 
-    recthread = threading.Thread(target = record_worker, 
-                        args = (stoprec, streamurl, target_dir, args), daemon = True)
-    recthread.start()
-    recthread.join(args.duration * 60)
+    # as long as recording is supposed to run
+    while time.time() < should_end_at:
+        stoprec = threading.Event()
+        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")
+        recthread.join(remaining)
+        verboseprint(print_time() + " ... Came out of rec thread again")
+
+        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()))
 
-    if(recthread.is_alive):
-        stoprec.set()
 
 def list(args):
-    settings = read_settings()
+    settings = read_settings(args)
     for key in sorted(settings['STATIONS']):
         print(key)
 
+
 def main():
     parser = argparse.ArgumentParser(description='This program records internet radio streams. '
-                                    'It is free software and comes with ABSOLUTELY NO WARRANTY.')
+                                                 '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, help='Name of the radio station '
-                                               '(see `radiorec.py list`)')
-    parser_record.add_argument('duration', type=check_duration, 
-                        help='Recording time in minutes')
-    parser_record.add_argument('name', nargs='?', type=str, 
-                        help='A name for the recording')
-    parser_record.add_argument('-p', '--public', action='store_true', help="Public write permissions (Linux only)")
-    parser_record.add_argument('-v', '--verbose', action='store_true', help="Verbose output")
+    parser_record.add_argument('station', type=str,
+                               help='Name of the radio station '
+                               '(see `radiorec.py list`)')
+    parser_record.add_argument('duration', type=check_duration,
+                               help='Recording time in minutes')
+    parser_record.add_argument('name', nargs='?', type=str,
+                               help='A name for the recording')
+    parser_record.add_argument(
+        '-p', '--public', action='store_true',
+        help="Public write permissions (Linux only)")
+    parser_record.add_argument(
+        '-v', '--verbose', action='store_true', help="Verbose output")
+    parser_record.add_argument(
+        '-s', '--settings', nargs='?', type=str,
+        help="specify alternative location for settings.ini")
     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")
 
+    if not len(sys.argv) > 1:
+        print('Error: No argument specified.\n')
+        parser.print_help()
+        sys.exit(1)
     args = parser.parse_args()
     args.func(args)
 
+
 if __name__ == '__main__':
     main()

settings.ini

@@ -1,11 +1,15 @@
 [GLOBAL]
 target_dir = $HOME/Arbeitsfläche
 
+
 [STATIONS]
 brklassik = http://streams.br-online.de/br-klassik_2.m3u
-dkultur = http://dradio_mp3_dkultur_m.akacast.akamaistream.net/7/530/142684/v1/gnl.akacast.akamaistream.net/dradio_mp3_dkultur_m
-dlf = http://dradio_mp3_dlf_m.akacast.akamaistream.net/7/249/142684/v1/gnl.akacast.akamaistream.net/dradio_mp3_dlf_m
+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
+ndrkultur = http://www.ndr.de/resources/metadaten/audio/m3u/ndrkultur.m3u