archiver.py

#!/usr/bin/python3

import cgi
import requests
import shutil
import os
import datetime
import schedule
import time
import traceback
import threading
# Api user/key from security tab in profile
ApiUser = os.environ["apiuser"]
ApiKey = os.environ["apikey"]

# Limit the size of the container to this, the site will not send you more torrents
# Once your total container size exceeds this amount. This is a 'soft' limit, allowing one torrent to
# exceed this amount. The site accepts single letter suffix B, K, M, G, T and P
ContainerSize = os.environ.get("size") or "1G"

# Directory to save the downloaded .torrent to
WatchDirectory = "/root/torrents"

# Limit the size of the container to this, the site will not send you more torrents
# Once your total container size exceeds this amount. This is a 'soft' limit, allowing one torrent to
# exceed this amount. The site accepts single letter suffix B, K, M, G, T and P

# Max 40 Character free form name. This is only used to define a container for the limit set above and will be used when
# sending you alerts regarding your container.
ContainerName = os.environ.get("name") or "mainContainer"

# The maximum number of partial/stalled torrents. 0 is infinite
MaxStalled = str(os.environ.get("stalled") or "0") 

# Base url
BaseURL = "https://passthepopcorn.me/"


# End configuration
headers = {
    'ApiUser': ApiUser,
    'ApiKey': ApiKey
}
payload = {
    'ContainerName': ContainerName,
    'ContainerSize': ContainerSize,
    'MaxStalled': MaxStalled
}

interval=int(os.environ.get("interval") or 20)


def run():
    try:
        now = datetime.datetime.now()
        print("Archive tool started: %s" % now.strftime("%Y-%m-%d %H:%M:%S"))
        print("Fetching json data")
        r = requests.get(BaseURL + 'archive.php?action=fetch', headers=headers, params=payload)
        r.raise_for_status()
        data = r.json()

        print("Parsing")

        # print(r.text)

        if data['Status'] == 'Ok':
            if 'TorrentID' not in data:
                raise ValueError("TorrentID not in json data!")

            payload2 = {
                'id': data['TorrentID'],
                'ArchiveID': data['ArchiveID']
            }
            print("Fetching torrent file for https://passthepopcorn.me/torrents.php?id=%s&torrentid=%s" % (data['GroupID'], data['TorrentID']))
            r = requests.get(BaseURL + 'torrents.php?action=download', headers=headers, params=payload2, stream=True)
            r.raise_for_status()

            params = cgi.parse_header(
                r.headers.get('Content-Disposition', ''))[-1]
            if 'filename' not in params:
                raise ValueError('Could not find a filename')

            filename = os.path.basename(params['filename'])
            abs_path = os.path.join(WatchDirectory, filename)
            with open(abs_path, 'wb') as target:
                r.raw.decode_content = True
                shutil.copyfileobj(r.raw, target)

            print("Saved %s" % abs_path)

        elif data['Status'] == 'Error':
            raise ValueError("Error: %s" % data['Error'])

        else:
            raise ValueError("Unexpected json result!")

        now = datetime.datetime.now()
        print("Archive tool finished: %s" % now.strftime("%Y-%m-%d %H:%M:%S"))
        print("")
    except Exception as E:
        print(traceback.format_exc())
        print(E)



def run_threaded(job_func):
    job_thread = threading.Thread(target=job_func)
    job_thread.start()

def job():
    schedule.every(interval).minutes.do(run_threaded,run)
    while True:
        schedule.run_pending()
        time.sleep(1)
if __name__ == "__main__":
    print("Starting script")
    job()