cdosea-play/cdoseaplay/update.py

#!/usr/bin/python3
from datetime import datetime
import json
import os
import requests
import shutil
import socket
import string
import sys
import time

import ox

socket.setdefaulttimeout(10)

from .config import base, lang

lock = '/tmp/update.lock'

prefix = os.path.expanduser('~/Videos/CDOSEA')
if not os.path.exists(prefix):
    os.makedirs(prefix)
os.chdir(prefix)

def get_subtitle(url, name):
    try:
        r = requests.get(url)
        if r.status_code == 200:
            with open('%s.tmp' % name, 'wb') as fd:
                fd.write(r.content)
            shutil.move('%s.tmp' % name, name)
        else:
            print('%s failed (%s)' % (url, r.status_code))
    except KeyboardInterrupt:
        os.unlink(lock)
        sys.exit(-1)
    except:
        pass

def get_all_subtitles():
    for i in range(10):
        for letter in string.ascii_lowercase:
            name = '%s%02d.%s.srt' % (letter, i, '1080p')
            rname = '%s%02d.%s.srt' % (letter, i, lang)
            url = '%s%s' % (base, rname)
            get_subtitle(url, name)

def update_video(url, name):
    folder = os.path.dirname(name)
    if folder and not os.path.exists(folder):
        os.makedirs(folder)
    if os.path.exists(name):
        try:
            head = requests.head(url, allow_redirects=True)
        except KeyboardInterrupt:
            sys.exit(-1)
        modified = head.headers.get("Last-Modified")
        if modified:
            mtime = time.mktime(datetime.strptime(modified, "%a, %d %b %Y %X GMT").timetuple())
            get = mtime > os.path.getmtime(name) or int(head.headers.get("Content-Length")) != os.path.getsize(name)
    else:
        get = True
    if get:
        print(url)
        try:
            r = requests.get(url, stream=True)
            if r.status_code == 200:
                with open('%s.tmp' % name, 'wb') as fd:
                    shutil.copyfileobj(r.raw, fd)
                ox.avinfo('%s.tmp' % name)['duration']
                shutil.move('%s.tmp' % name, name)
                mtime = time.mktime(datetime.strptime(r.headers.get("Last-Modified"), "%a, %d %b %Y %X GMT").timetuple())
                os.utime(name, (mtime, mtime))
                ox.avinfo(name)['duration']
                get_subtitle(url.replace('1080p.mp4', lang + '.srt'), name.replace('.mp4', '.srt'))
            else:
                raise Exception('failed to download %s (%s)' % (url, r.status_code))
        except KeyboardInterrupt:
            os.unlink(lock)
            sys.exit(-1)
        except:
            print(url, 'failed')

def update_videos():
    if os.path.exists(lock):
        print('another update is already running')
        sys.exit(0)
    with open(lock, 'w') as f:
        f.write('')
    for i in range(10):
        for letter in string.ascii_lowercase:
            name = '%s%02d.1080p.mp4' % (letter, i)
            url = '%s%s' % (base, name)
            update_video(url, name)
    os.unlink(lock)


if __name__ == '__main__':
    if len(sys.argv) > 1 and sys.argv[1] == 'subtitles':
        get_all_subtitles()
    else:
        update_videos()