diff --git a/TorrentToMedia.cfg.sample b/TorrentToMedia.cfg.sample new file mode 100644 index 00000000..e1e4c646 --- /dev/null +++ b/TorrentToMedia.cfg.sample @@ -0,0 +1,16 @@ +[CouchPotato] +category = movies +destination = /abs/path/to/complete/movies + +[SickBeard] +category = tv +destination = /abs/path/to/complete/tv + +[Torrent] +uselink = 0 +extractiontool = 'C:\\Program Files\\7-Zip\\7z.exe' + +[Notes] +copy = 0 will create hard links to your original download. +If moving to new hard-disk or network mount, use copy = 1 +copy = 0 is much faster and uses less disk space. diff --git a/TorrentToMedia.py b/TorrentToMedia.py new file mode 100644 index 00000000..07342601 --- /dev/null +++ b/TorrentToMedia.py @@ -0,0 +1,161 @@ +#!/usr/bin/env python + +import autoProcessMovie +import autoProcessTV +import sys, os, glob, ConfigParser +from os import listdir +from os.path import isfile, join + +##You can use the following parameters (UTORRENT): +## +##%F - Name of downloaded file (for single file torrents) +##%D - Directory where files are saved +##%N - Title of torrent +##%P - Previous state of torrent +##%L - Label +##%T - Tracker +##%M - Status message string (same as status column) +##%I - hex encoded info-hash +##%S - State of torrent +##%K - kind of torrent (single|multi) +## +##Where State is one of: +## +##Error - 1 +##Checked - 2 +##Paused - 3 +##Super seeding - 4 +##Seeding - 5 +##Downloading - 6 +##Super seed [F] - 7 +##Seeding [F] - 8 +##Downloading [F] - 9 +##Queued seed - 10 +##Finished - 11 +##Queued - 12 +##Stopped - 13 + +## We will pass in %D, %N, %L from uTorrent +Directory = sys.argv[1] ## %D -- Example output: F:\path\to\dir\My.Series.S01E01.720p.HDTV.x264-2HD +Name = sys.argv[2] ## %N -- Example output: My.Series.S01E01.720p.HDTV.x264-2HD +Category = sys.argv[3] ## %L -- Example output: tvseries ## This is the label in uTorrent + +status = 0 +packed = 0 + +config = ConfigParser.ConfigParser() +configFilename = os.path.join(os.path.dirname(sys.argv[0]), "TorrentToMedia.cfg") + +print "torrentToMedia v 4.1" +print "INFO: Loading config from", configFilename + +if not os.path.isfile(configFilename): + print "ERROR: You need an autoProcessMovie.cfg file - did you rename and edit the .sample?" + sys.exit(-1) + +config.read(configFilename) + +TV_Cat = config.get("SickBeard", "category") +TV_dest = config.get("CouchPotato", "destination") +Movie_dest = config.get("CouchPotato", "destination") +Movie_Cat = config.get("CouchPotato", "category") +useLink = int(config.get("Torrent", "uselink")) +extractionTool = config.get("Torrent", "extractiontool") + +if Category == Movie_Cat: + destination = Movie_dest +elif Category == TV_Cat: + destination = TV_dest +else; + print "INFO: Not assigned a label of either", Movie_Cat, "or", TV_Cat, ". Exiting" + +test = re.compile('^(.*)\.((zip|rar|7z|gz|bz|tar|arj)|(r[0-9]{1,3})|([0-9]{1,3}))$', re.IGNORECASE|re.UNICODE); +if test.match(Directory): + print "INFO: Found compressed archives, extracting" + packed = 1 + +## TODO: Check that files actully is .mkv / .avi etc, and not packed files or anything else +if useLink == 1 and packed == 0: ## symlinks + print "INFO: Copying all files from", Directory, "to", destination + shutil.copytree(Directory, destination) + +elif useLink == 0 and packed == 0: ## hardlink + print "INFO: Creating hard link from", Directory, "to", destination + shutil.copytree(src, dst, copy_function=os.link) + +elif useLink == 0 and packed == 1: ## unpack + ## Using Windows? + if os.name == 'nt': + cmd_7zip = [extractionTool, 'x -y'] + ext_7zip = [".rar",".zip",".tar.gz","tgz",".tar.bz2",".tbz",".tar.lzma",".tlz",".7z",".xz"] + EXTRACT_COMMANDS = dict.fromkeys(ext_zip, cmd_7zip) + print "INFO: We are using Windows" + + ## Using linux? + elif os.name == 'posix': + required_cmds=["unrar", "unzip", "tar", "unxz", "unlzma", "7zr"] + EXTRACT_COMMANDS = { + ".rar": ["unrar", "x -o+ -y"], + ".zip": ["unzip", ""], + ".tar.gz": ["tar", "xzf"], + ".tgz": ["tar", "xzf"], + ".tar.bz2": ["tar", "xjf"], + ".tbz": ["tar", "xjf"], + ".tar.lzma": ["tar", "--lzma xf"], + ".tlz": ["tar", "--lzma xf"], + ".txz": ["tar", "--xz xf"], + ".7z": ["7zr", "x"], + } + print "INFO: We are using *nix" + + ## Need to add a check for which commands that can be utilized in *nix systems.. + else: + print "ERROR: Unknown OS, exiting" + + files = [ f for f in listdir(Directory) if isfile(join(Directory,f)) ] + for f in files: + ext = os.path.splitext(f["path"]) + if ext[1] in (".gz", ".bz2", ".lzma"): + ## Check if this is a tar + if os.path.splitext(ext[0]) == ".tar": + cmd = EXTRACT_COMMANDS[".tar" + ext[1]] + else: + if ext[1] in EXTRACT_COMMANDS: + cmd = EXTRACT_COMMANDS[ext[1]] + else: + print("ERROR: Unknown file type: %s", ext[1]) + continue + fp = os.path.join(destination, os.path.normpath(f["path"])) + + ## Destination path + dest = os.path.join(destination, Name) + + ## Create destionation folder + if not os.path.exists(dest): + try: + os.makedirs(dest) + except Exception, e: + print("ERROR: Not possible to create destination folder: %s", e) + return + + print("INFO: Extracting to %s", dest) + def on_extract_success(result): + print("INFO: Extraction was successful for %s") + + def on_extract_failed(result): + print("ERROR: Extraction failed for %s") + + ## Running.. + print("INFO: Extracting %s %s %s %s", cmd[0], cmd[1], fp, dest) + d = getProcessValue(cmd[0], cmd[1].split() + [str(fp)], {}, str(dest)) + d.addCallback(on_extract_success) + d.addErrback(on_extract_failed) +else: + print "INFO: Didn't find any compressed archives or media files to process, exiting" + +status = int(status) +## Now we pass off to CouchPotato or SickBeard. +if Category == Movie_Cat: + autoProcessMovie.process(destination, Name, status) +elif Category == TV_Cat: + autoProcessTV.processEpisode(destination, Name, status)