martin/radiorec
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Improved content type handling

Browse source
This commit is contained in:
Martin Brodbeck 2013-09-16 15:49:35 +02:00
parent 6d6312f2d6
commit 190a9ec263
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
radiorec.py
View file

@ -41,8 +41,14 @@ def record_worker(stoprec, streamurl, target_dir, name=None):
filename = target_dir + os.sep + datetime.datetime.now().isoformat() filename = target_dir + os.sep + datetime.datetime.now().isoformat()
if name: if name:
filename += '_' + name filename += '_' + name
if(conn.getheader('Content-Type') == 'audio/mpeg'): content_type = conn.getheader('Content-Type')
if(content_type == 'audio/mpeg'):
filename += '.mp3' filename += '.mp3'
elif(content_type == 'application/ogg' or content_type == 'audio/ogg'):
filename += '.ogg'
else:
print('Unknown content type. Assuming mp3.')
filename += 'mp3'
target = open(filename, "wb") target = open(filename, "wb")
while(not stoprec.is_set() and not conn.closed): while(not stoprec.is_set() and not conn.closed):
target.write(conn.read(1024)) target.write(conn.read(1024))