From 48c8b44bd3350287846dc0490ec090a0dc661d0f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joel=20K=C3=A5berg?= <joel.kaberg@gmail.com>
Date: Mon, 25 Feb 2013 11:02:51 +0100
Subject: [PATCH] hardlink solution for uTorrent

---
 TorrentToMedia.py           |  27 +++++++-
 autoProcessMedia.cfg.sample |   4 ++
 utorrent/__init__.py        |   0
 utorrent/__init__.pyc       | Bin 0 -> 133 bytes
 utorrent/client.py          | 124 ++++++++++++++++++++++++++++++++++++
 utorrent/client.pyc         | Bin 0 -> 5325 bytes
 utorrent/upload.py          |  71 +++++++++++++++++++++
 utorrent/upload.pyc         | Bin 0 -> 2798 bytes
 8 files changed, 224 insertions(+), 2 deletions(-)
 create mode 100644 utorrent/__init__.py
 create mode 100644 utorrent/__init__.pyc
 create mode 100644 utorrent/client.py
 create mode 100644 utorrent/client.pyc
 create mode 100644 utorrent/upload.py
 create mode 100644 utorrent/upload.pyc

diff --git a/TorrentToMedia.py b/TorrentToMedia.py
index 3e990b56..6c759cfa 100755
--- a/TorrentToMedia.py
+++ b/TorrentToMedia.py
@@ -7,14 +7,18 @@ import os
 import shutil
 import logging
 import datetime
+import time
+import urllib2
 from subprocess import call
 
+
 # Custom imports
 import linktastic.linktastic as linktastic
 import autoProcessMovie
 import autoProcessTV
 from nzbToMediaEnv import *
 from nzbToMediaUtil import *
+from utorrent.client import UTorrentClient
 
 nzbtomedia_configure_logging(os.path.dirname(sys.argv[0]))      
 Logger = logging.getLogger(__name__)
@@ -272,7 +276,7 @@ if len(sys.argv) == 2: #for other clients we assume we must at least get the dir
         inputName = '' # We dont have a name yet
         inputCategory = '' # We dont have a category yet
 
-elif len(sys.argv) > 3 and sys.argv[1] == 'utorrent': #distinguish utorrent from others like deluge.
+elif len(sys.argv) > 4 and sys.argv[1] == 'utorrent': #distinguish utorrent from others like deluge.
         # We will pass in 'utorrent' '%D', '%N', and '%L' (if it exists), from uTorrent
         # In short pass "/path/to/downloaded/torrent/ name" to TorrentToMedia.py, eg  >>>> TorrentToMedia.py /Downloaded/MovieName.2013.BluRay.1080p.x264-10bit.DTS MovieName.2013.BluRay.1080p.x264-10bit.DTS <<<<
         inputDirectory = os.path.normpath(sys.argv[2])
@@ -281,6 +285,7 @@ elif len(sys.argv) > 3 and sys.argv[1] == 'utorrent': #distinguish utorrent from
                 inputCategory = sys.argv[4] # We dont have a category yet
         except:
                 inputCategory = '' # We dont have a category yet
+        inputHash = sys.argv[5]
 
 elif len(sys.argv) == 4:
         # We will assume this to be the passin from deluge. torrent id, torrent name, torrent save path.
@@ -312,6 +317,9 @@ movieDestination = os.path.normpath(config.get("CouchPotato", "outputDirectory")
 # Torrent specific
 useLink = int(config.get("Torrent", "useLink"))
 extractionTool = os.path.normpath(config.get("Torrent", "extractionTool"))
+uTorrentWEBui = config.get("Torrent", "uTorrentWEBui")
+uTorrentUSR = config.get("Torrent", "uTorrentUSR")
+uTorrentPWD = config.get("Torrent", "uTorrentPWD")
 compressedContainer = (config.get("Torrent", "compressedExtentions")).split(',')
 mediaContainer = (config.get("Torrent", "mediaExtentions")).split(',')
 metaContainer = (config.get("Torrent", "metaExtentions")).split(',')
@@ -429,11 +437,26 @@ old_stdout = sys.stdout  # Still crude, but we wat to capture this for now
 logFile = os.path.join(os.path.dirname(sys.argv[0]), "postprocess.log")
 log_file = open(logFile,"a+")
 sys.stdout = log_file
-if inputCategory == movieCategory:  
+
+# Hardlink solution with uTorrent
+if inputHash and useLink:  
+    Logger.debug("MAIN: We are using hardlinks with uTorrent, calling uTorrent to pause download")
+    utorrentClass = UTorrentClient(uTorrentWEBui, uTorrentUSR, uTorrentPWD)
+    utorrentClass.stop(inputHash)
+    time.sleep(10)
+
+if inputCategory == movieCategory:
         Logger.info("MAIN: Calling postprocessing script for CouchPotatoServer")
         autoProcessMovie.process(outputDestination, inputName, status)
 elif inputCategory == tvCategory:
         Logger.info("MAIN: Calling postprocessing script for Sick-Beard")
         autoProcessTV.processEpisode(outputDestination, inputName, status)
+
+# Hardlink solution with uTorrent
+if inputHash and useLink:
+    Logger.debug("MAIN: We are using hardlinks with uTorrent, calling uTorrent to resume download")
+    utorrentClass.start(inputHash)
+    time.sleep(10)
+            
 sys.stdout = old_stdout
 log_file.close()
diff --git a/autoProcessMedia.cfg.sample b/autoProcessMedia.cfg.sample
index f20ff807..af847ca3 100644
--- a/autoProcessMedia.cfg.sample
+++ b/autoProcessMedia.cfg.sample
@@ -31,6 +31,10 @@ failed_fork=0
 useLink = 0
 extractionTool = C:\Program Files\7-Zip\7z.exe
 categories = music,music_videos,pictures,software
+###### uTorrent Hardlink solution (You must edit this if your using TorrentToMedia.py with uTorrent)
+uTorrentWEBui = http://localhost:8090/gui/
+uTorrentUSR = your username
+uTorrentPWD = your password
 ###### ADVANCED USE - ONLY EDIT IF YOU KNOW WHAT YOU'RE DOING ######
 compressedExtentions = .zip,.rar,.7z,.gz,.bz,.tar,.arj
 mediaExtentions = .mkv,.avi,.divx,.xvid,.mov,.wmv,.mp4,.mpg,.mpeg,.vob,.iso
diff --git a/utorrent/__init__.py b/utorrent/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/utorrent/__init__.pyc b/utorrent/__init__.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..d4de8d258fa8be36a8f3a88f5e62faa47807b69c
GIT binary patch
literal 133
zcmZSn%*(Y;Ry#160SXv_v;z<qvjB+{28Lh_kcgiKkYGR~iuHhE&Q>ul`Q>>z`H3m1
zMa40BRY@WFzNsmhi7}-m`9(#kc_lIN@tJv<CGqik1(hWnKrJ@8`6;D2sdgZPi-DK{
E0QV~#1poj5

literal 0
HcmV?d00001

diff --git a/utorrent/client.py b/utorrent/client.py
new file mode 100644
index 00000000..baffbf6e
--- /dev/null
+++ b/utorrent/client.py
@@ -0,0 +1,124 @@
+#coding=utf8
+import urllib
+import urllib2
+import urlparse
+import cookielib
+import re
+import StringIO
+try:
+    import json 
+except ImportError:
+    import simplejson as json
+
+from upload import MultiPartForm
+
+class UTorrentClient(object):
+
+    def __init__(self, base_url, username, password):
+        self.base_url = base_url
+        self.username = username
+        self.password = password
+        self.opener = self._make_opener('uTorrent', base_url, username, password)
+        self.token = self._get_token()
+        #TODO refresh token, when necessary
+
+    def _make_opener(self, realm, base_url, username, password):
+        '''uTorrent API need HTTP Basic Auth and cookie support for token verify.'''
+
+        auth_handler = urllib2.HTTPBasicAuthHandler()
+        auth_handler.add_password(realm=realm,
+                                  uri=base_url,
+                                  user=username,
+                                  passwd=password)
+        opener = urllib2.build_opener(auth_handler)
+        urllib2.install_opener(opener)     
+
+        cookie_jar = cookielib.CookieJar()
+        cookie_handler = urllib2.HTTPCookieProcessor(cookie_jar)
+
+        handlers = [auth_handler, cookie_handler]
+        opener = urllib2.build_opener(*handlers)
+        return opener
+
+    def _get_token(self):
+        url = urlparse.urljoin(self.base_url, 'token.html')
+        response = self.opener.open(url)
+        token_re = "<div id='token' style='display:none;'>([^<>]+)</div>"
+        match = re.search(token_re, response.read())
+        return match.group(1)
+
+       
+    def list(self, **kwargs):
+        params = [('list', '1')]
+        params += kwargs.items()
+        return self._action(params)
+
+    def start(self, *hashes):
+        params = [('action', 'start'),]
+        for hash in hashes:
+            params.append(('hash', hash))
+        return self._action(params)
+        
+    def stop(self, *hashes):
+        params = [('action', 'stop'),]
+        for hash in hashes:
+            params.append(('hash', hash))
+        return self._action(params)
+ 
+    def pause(self, *hashes):
+        params = [('action', 'pause'),]
+        for hash in hashes:
+            params.append(('hash', hash))
+        return self._action(params)
+ 
+    def forcestart(self, *hashes):
+        params = [('action', 'forcestart'),]
+        for hash in hashes:
+            params.append(('hash', hash))
+        return self._action(params)
+ 
+    def getfiles(self, hash):
+        params = [('action', 'getfiles'), ('hash', hash)]
+        return self._action(params)
+ 
+    def getprops(self, hash):
+        params = [('action', 'getprops'), ('hash', hash)]
+        return self._action(params)
+        
+    def setprio(self, hash, priority, *files):
+        params = [('action', 'setprio'), ('hash', hash), ('p', str(priority))]
+        for file_index in files:
+            params.append(('f', str(file_index)))
+
+        return self._action(params)
+        
+    def addfile(self, filename, filepath=None, bytes=None):
+        params = [('action', 'add-file')]
+
+        form = MultiPartForm()
+        if filepath is not None:
+            file_handler = open(filepath)
+        else:
+            file_handler = StringIO.StringIO(bytes)
+            
+        form.add_file('torrent_file', filename.encode('utf-8'), file_handler)
+
+        return self._action(params, str(form), form.get_content_type())
+
+    def _action(self, params, body=None, content_type=None):
+        #about token, see https://github.com/bittorrent/webui/wiki/TokenSystem
+        url = self.base_url + '?token=' + self.token + '&' + urllib.urlencode(params)
+        request = urllib2.Request(url)
+
+        if body:
+            request.add_data(body)
+            request.add_header('Content-length', len(body))
+        if content_type:
+            request.add_header('Content-type', content_type)
+
+        try:
+            response = self.opener.open(request)
+            return response.code, json.loads(response.read())
+        except urllib2.HTTPError,e:
+            raise 
+        
diff --git a/utorrent/client.pyc b/utorrent/client.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..840aafc0f35d4188671d0200e9f2f937013e1b0f
GIT binary patch
literal 5325
zcmcgwYjYe&746wwt+aYrA|WV-n5+wkmKZt2Jd!fD@{kEps0c9<<sfW^(auOatJ#^&
zbZ_LP<S*D?sNz@fL-@!SehxnX&N;m+%T9_aqQu(mR=@k+b5GyCz4d=D&Hv-dwcmEl
z<ZlN5-$k;2N0#6lnB18CvvOeegK2xs?ANB{y4kN!%MG*Nn3iYE{)`!~Osh?^-!ylN
zE2fdon&Wv>otPj9Ci~2>vk1&5tk+E5GJ|=uKW~DhZqfyFh>aWOu`!Q~`ThPPMceIA
z-t0g4+wKG?x+P>Ek8+po#FhJ?tOo89T7I`zR#jTK%{)VXrfeNu?<3j2B1@sBRKpI`
zE8tdCs-^-os;e+Fkpi+NGisSNNmGS6lgz5nGRd3@^CoGjuwasT6(IMLz6<6t0K7D<
zE~>h0!rcxf?lzDyQNc|lyM^pDFsH^uDAZIywXTAyeF%#dd{M2NM-6o;@6Ql5&V|$l
zxYe-QwwpsfnqbqJMMW(Ku}z~<mHTd+Rz*BWRT;+Co|IMM7BMy&#K&n=4$~s7ToVnG
z;yn9~QWv@M7-&-gl{!?%F|%oY7!v%DwU8q^NItVaLTU5ny*uSek(Y6jR`y=;=YzfS
z<21?Qdn4yfd9N#5ULTGnSrlbO=A!5um~1J530nGmC+Kp0Hnorh<zp0_G5vs`Je)D7
zGo~+#13Q=-@WIwVFadTAjA;ZMjV7QNn1@YGMqiWF02=mD!b-F1m^G)f8aKxYr*r10
zW*)ZGg(X_%Zy4F+WY1pv0i>B4q5a;@R=Y^kr2XOE-cI}d*k;}Kdn4Ct$3@camgRAl
zw(V#*EGySOEUUJVZGV<l+2Q!Qu&&ZLA2^~rsxrYnf(d0HnJ?+k>W<cijq~i_hWjSU
z+)vwaw-2#fj!86goFvgxc0U+pc`~&-xtbN$#d$t$BFtV=Ebaul<o*aJ^2Nzcn`+sq
z%5G|HS#@ZhP#L#;h8tcCh!yp`T%oAS;eMZ}kEL~LwCr@4h%uRBFF>%6<i3n#Uq)u?
zi@|cx#Ag=&mujdn1zq|dQjO4WBH0j`rx^~OC)guUL1!3j>VrXGAP5D%p}lTAlX#?S
zrVkFm)uCM^?8CR<NYmx<%xsI}GXC|R8|3z<XuOqVpS82(_L|z)+SZNp^!8ek*<l`!
zZx&^d{$g#T^M^m)+PHu1hqvCq@QsX8cW5Mk+N*pRS2mr}qhDr)w}v3+);#5o?xRXY
zDx1btw<n^3dWkrARF$Kle6x%?VWW2_RceQ2fh|Nsx`+rX5x7C@y1k3oi7Q@1nmB-$
zf)`+=m%U|Z6-xbU_P>S<a2^SJddtA{b+5byThLhYp-<43c6nx<BV2FFds>3QwvZyL
zNm^ts9a!NP#a)+`MTc8^97v?&lejvvlk-@`g9|Zp4GaDR$#P}7ycMr4Wn=mklD&xx
zuFPxu;3~DfbN@04N1h1@aPF`sZ}#bmx%)H%G|jf;_4-IlocPKKmR@Xo9uUglhgi<h
z5RZ#`h9FK7zK$MI_z5ChA+;g>e)tBvk-p?z5MIaBw~;KlKtPs))vpPXa2-1c0_)15
zmz)ffEhae72J96K;g3;!27vHQwtXe4-@r;6-0yj*P9qOv+%8|N`&sw%P@ae2&hu2B
zw+}90++REiUyQ)hW1a^Lyaqn{91LV7oe<L|dtXHcj{p}4Ux0(p2*o=-g_d%BI`*F9
zBoqR=!z{-!)d_u;=jTT#5cf1c^L&~mns>0)mp$A+!_*xlxn2YB-u;R+8&>7;e`}@@
zU!P__hgN$}X-4SiHfaWV@lQ~|JLAk|upfd8cyL&YCag13c(BH}_0IiQ71lO*k>QcT
zaSC0hKEC69JWP)8mN5ly_ib-%7I8ZdtE`mObma||ABw`~{oA?0y5@A#TUVWPbyUKK
zc4UrjeXfnzbgY}ivx*+j>AZksh~<;?!9|$z7M3Gm6JK~0*T9vU-c*#BURmeh1O?#3
zn^9;!^3MUpW4NbVrdl_MbUefe*^6ZZ{rC=0B*q{fq5t$4;fV#408Ks%>L%bNiV;{B
z4+5dj(Q^iSpnJDNTs-6w5@Gf_p-`|pdE-RFh71_F!}Yg6hdxi;be>{3Mx6di+D5P!
zewQ^4+I3Y{9ButZ6$bR$i4vqmw@gx<){t?6TS3OhTemD60=>wMhpFh{?`y8c$!AYO
zR)(?b$(9FWH@V1lFTy_#qz?y8qw))wRZ$hh*K!cVGUmr6Uuy-7!k25WcpFnlq9z;v
z3kBK&mJZCS4{cTvZ?D3%w53AuRoWlN1@AT4kdMNess1Dj!-_sOq5<*PN1%?UothAz
z2L;xzn`*4F=tjH_e)x{@4ei}B{rL$A{L`<$U;s>iiDc<t5+`(uckdRLO@opw>*tYw
zngFlL;4fzSyNG`?Zd=m$9mV;#bu?cS7MtE^>#)y}>sh{~rcJUVjqAgCMf@aye*@~7
zhT?<{94D~D^wSY;xHH})v5R#~NunOYeLNED7>tQ}lzGbWe34cnLEl0sT7!>y8(!to
z+?@~3Use?!e>9i(5b^??eNZN2G3PuD!|!n}HLtu_;rH3VkUYHrt3>`Bi}>eV?FXPz
z6At4t9uF_p>bPdwNZTEOLRq7T?2jS=5k-SC8G+tWxPU&r6+>yX#wz2{kdax)2qonG
z9P%0q>G4Clsqh_UTg+ZZ_Cyht{_kHyLc+0(WWSB9S!>O;UO<{@H7B1&tI?WWUh43d
zvXHN#P^@@`rI(m7I0)ZmMt9<<)z*MNILKO3+Gc}cp31V~(2%zE&WH8#pr3Y~B)O<M
dV%{we@rI&~Z()N)ocHOcjyKOzuu@y8{TC0I8bAO5

literal 0
HcmV?d00001

diff --git a/utorrent/upload.py b/utorrent/upload.py
new file mode 100644
index 00000000..e7c06f40
--- /dev/null
+++ b/utorrent/upload.py
@@ -0,0 +1,71 @@
+#code copied from http://www.doughellmann.com/PyMOTW/urllib2/
+
+import itertools
+import mimetools
+import mimetypes
+from cStringIO import StringIO
+import urllib
+import urllib2
+
+class MultiPartForm(object):
+    """Accumulate the data to be used when posting a form."""
+
+    def __init__(self):
+        self.form_fields = []
+        self.files = []
+        self.boundary = mimetools.choose_boundary()
+        return
+    
+    def get_content_type(self):
+        return 'multipart/form-data; boundary=%s' % self.boundary
+
+    def add_field(self, name, value):
+        """Add a simple field to the form data."""
+        self.form_fields.append((name, value))
+        return
+
+    def add_file(self, fieldname, filename, fileHandle, mimetype=None):
+        """Add a file to be uploaded."""
+        body = fileHandle.read()
+        if mimetype is None:
+            mimetype = mimetypes.guess_type(filename)[0] or 'application/octet-stream'
+        self.files.append((fieldname, filename, mimetype, body))
+        return
+    
+    def __str__(self):
+        """Return a string representing the form data, including attached files."""
+        # Build a list of lists, each containing "lines" of the
+        # request.  Each part is separated by a boundary string.
+        # Once the list is built, return a string where each
+        # line is separated by '\r\n'.  
+        parts = []
+        part_boundary = '--' + self.boundary
+        
+        # Add the form fields
+        parts.extend(
+            [ part_boundary,
+              'Content-Disposition: form-data; name="%s"' % name,
+              '',
+              value,
+            ]
+            for name, value in self.form_fields
+            )
+        
+        # Add the files to upload
+        parts.extend(
+            [ part_boundary,
+              'Content-Disposition: file; name="%s"; filename="%s"' % \
+                 (field_name, filename),
+              'Content-Type: %s' % content_type,
+              '',
+              body,
+            ]
+            for field_name, filename, content_type, body in self.files
+            )
+        
+        # Flatten the list and add closing boundary marker,
+        # then return CR+LF separated data
+        flattened = list(itertools.chain(*parts))
+        flattened.append('--' + self.boundary + '--')
+        flattened.append('')
+        return '\r\n'.join(flattened)
diff --git a/utorrent/upload.pyc b/utorrent/upload.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..5b259013fd586c94ce38ae8275ee58627dff9964
GIT binary patch
literal 2798
zcmb_e+in|G6x}mpJ5EU2N{FQ8B8EaFqc$#rKA_%2X+ZE$1Ct5~D`+&CId;a3XUv>q
z+M?vCh>OG*@G<-p9{_9Zu}v#@DIv~mpZi&R?Y-AH_+xA1_t&m`k;>vP!1o7u><A{q
zzbA#raMku?=q<;68T#7xr3hr&l3`1{#h$hd+skzw8FnNK#H`5(n_KdO$P<yThFu<i
z&}-wW{7?K1dLEjt3v=M=yc~b}ncG6={=9Jc=Sl58uIlLus*jlu@z^^s1|Ss}A$h!E
z7)a)8qs7ZP0z?3U`{Bi!WNjVnN_5a$<2moa+)dN@bY3LRgzngcS>lq=RpF5d=hkH5
z!(&s1v&uSv3zKjJJpGiQ`4G@OJa!jmEOH_euNs~>UnT(^=QuFIN%eQ21h^NliRZ<m
z+!UF0WMz~W#=3R1ruo#ksw%8Ik6wCQRo28u)x6A-`mr*$>@*y<u*#ZZ)Wf%^u~*T$
zcjIusdRP`!l9}2bmfs&esP3CAPY&m<s%ula!}*M}{aJHr9Oq^3;+SEu42*c|_?-9t
z=`|ey_3&;YvVG7a9RnM}f>4*w5Wj15aF3+k43~Is?^=Z%b+EE&F&bw9egXT(#>Ht>
zI=B?O$20RH#w;O;mj<xQM+5dU1{&NGIDmg&M!r1pkj_j{ZnJN}+|9Bu32i=|6(-a;
zF*%qhbWF3QKhWh7_c*$g%x0#{db}^9V~(LI6NHly{w^uz<}AK4Ig%`EvSu3{%k#zG
z_EMft1bo!Z2WXrG49^ph)3%&wWqi|>`uZX};W)g8)0bv_md|NR%7&VjOt3+z!M2mO
zj9Lvr0SMA;BZ&<`OxtDEXf~OwpJP_+MHmoL<Z0sas=QXE&bU46YLiR{JpxsF(9N_l
zs7mv9{v(XmZD7%OZmiX?M#?M4ba%rrInn4S&CFUN(hgU$^GQ->h0(&%(!$Yt;Zc=6
zK1(=wSrd-Jyn?pnh2n2|o8GJVXw?w%^0B|61Fbk!)|(vM`301Ff>itoU@_s#k4+GK
z*++Pom(R=x%lTk64`+~HlMWC174i1u)Mr{xpnJ<2-9c+d4{OF30mZg1mty8XNwpsV
zV`SXCE<<F$DsxzyS#2z;i%R%R1zZjDGA-trN~Cj1dJN%ICF^Se?(L;iOESO=rRv%%
zFmV3sH{=jWr(l29cMjlh6R<DimVD`btD=6O)nwnsrh83^@9pOnvYa!iZfG$sRE9yn
zeR<bjc33QfCbYa7YAn)!C!%Uan=p-TQs<~tD{a*s&#j%7DR^saO7mz|-|3^Hu6Nq%
z`m!+9b-umm`gJ*MEnGhhSa>S};5^hwX$bIzIJRWmUa+w$KOXk)Y#`llhASd&YAjY=
zOR-$;dw?>y5$@WEb-EynV53VgT6tPlF?UB)R7d8Xmf46B)TlW1;QwIneVo&<x2CwX
zy|vLJ#ro1vnKh3f^_j}9YJygCkz1$okvmg2uTQ#@9w&Jju`+elq{_>u^qDJ|0?mte
zgx>-<*<?-XFmoe%_S`AqDE@Jb@`>X&c>p<<PRH8{I^JgcTnl#XT>B;OqPOJ@)H{`@
zSlNzar9Y0RRW>gSy3KJ}mBw+@#YvI!!TgQ*6Cv86;Y+Ilj>dnCMns=UmSwOzUdLbW
ztOx7u&2AHd*D)IL4G=Mvl<su-^DV+i^BSDjMUfvV>ZbXwhOIi9nAGVNXT_;(-D>!`
P!xC4#RNhVR0<HfS_aaw%

literal 0
HcmV?d00001