change code style

Author: PeterDing

115.py

@@ -32,7 +32,13 @@
 ############################################################
 
 # file extensions
-mediatype = {".wma", ".wav", ".mp3", ".aac", ".ra", ".ram", ".mp2", ".ogg", ".aif", ".mpega", ".amr", ".mid", ".midi", ".m4a", ".m4v", ".wmv", ".rmvb", ".mpeg4", ".mpeg2", ".flv", ".avi", ".3gp", ".mpga", ".qt", ".rm", ".wmz", ".wmd", ".wvx", ".wmx", ".wm", ".swf", ".mpg", ".mp4", ".mkv", ".mpeg", ".mov", ".mdf", ".iso", ".asf"}
+mediatype = {
+    ".wma", ".wav", ".mp3", ".aac", ".ra", ".ram", ".mp2", ".ogg", ".aif",
+    ".mpega", ".amr", ".mid", ".midi", ".m4a", ".m4v", ".wmv", ".rmvb",
+    ".mpeg4", ".mpeg2", ".flv", ".avi", ".3gp", ".mpga", ".qt", ".rm",
+    ".wmz", ".wmd", ".wvx", ".wmx", ".wm", ".swf", ".mpg", ".mp4", ".mkv",
+    ".mpeg", ".mov", ".mdf", ".iso", ".asf"
+}
 
 s = '\x1b[%d;%dm%s\x1b[0m'       # terminual color template
 
@@ -121,8 +127,9 @@ def get_ssopw(ssoext):
             "goto=http://m.115.com/?ac=home"
 
         theaders = headers
-        theaders["Referer"] = "http://passport.115.com/static/reg_login_130418/bridge.html?ajax_cb_key=bridge_%s" \
-        % int(time.time()*1000)
+        theaders["Referer"] = "http://passport.115.com\
+            /static/reg_login_130418/bridge.html?ajax_cb_key=bridge_%s" \
+            % int(time.time()*1000)
 
         # Post!
         # XXX : do not handle errors
@@ -190,8 +197,11 @@ def get_infos(self, cid):
                 j = ss.get(url, params=params).json()
                 if j['errNo'] == 0 and j['data']:
                     if args.type_:
-                        j['data'] = [x for x in j['data'] if x.get('ns') \
-                            or x['ico'].lower() == unicode(args.type_.lower())]
+                        j['data'] = [
+                            x for x in j['data'] \
+                            if x.get('ns') \
+                                or x['ico'].lower() == unicode(args.type_.lower())
+                        ]
 
                     for i in j['data']:
                         if i.get('ns'):
@@ -202,12 +212,17 @@ def get_infos(self, cid):
                             dir_loop2.append(item)
 
                     if args.play:
-                        j['data'] = [i for i in j['data'] \
-                            if i.get('sha') and os.path.splitext(i['n'])[-1].lower() in mediatype]
+                        j['data'] = [
+                            i for i in j['data'] \
+                            if i.get('sha') \
+                                and os.path.splitext(i['n'])[-1].lower() \
+                                in mediatype
+                        ]
 
                     total_file = len([i for i in j['data'] if not i.get('ns')])
                     if args.from_ - 1:
-                        j['data'] = j['data'][args.from_-1:] if args.from_ else j['data']
+                        j['data'] = j['data'][args.from_-1:] if args.from_ \
+                                                                else j['data']
                     nn = args.from_
                     for i in j['data']:
                         if not i.get('ns'):
@@ -219,8 +234,12 @@ def get_infos(self, cid):
                                 'dir_': os.path.split(t)[0],
                                 'dlink': self.get_dlink(i['pc']),
                                 'name': i['n'].encode('utf8'),
-                                #'purl': self._get_play_purl(i['pc'].encode('utf8')) if args.play and self.is_vip else None,
-                                'purl': self._get_play_purl(i['pc'].encode('utf8')) if args.play else None,
+                                #'purl': self._get_play_purl(
+                                #   i['pc'].encode('utf8')) \
+                                #       if args.play and self.is_vip else None,
+                                'purl': self._get_play_purl(
+                                    i['pc'].encode('utf8')) \
+                                        if args.play else None,
                                 'nn': nn,
                                 'total_file': total_file
                             }
@@ -245,12 +264,17 @@ def download(infos):
         num = random.randint(0, 7) % 7
         col = s % (2, num + 90, infos['file'])
         infos['nn'] = infos['nn'] if infos.get('nn') else 1
-        infos['total_file'] = infos['total_file'] if infos.get('total_file') else 1
-        print '\n  ++ 正在下载: #', s % (1, 97, infos['nn']), '/', s % (1, 97, infos['total_file']), '#', col
+        infos['total_file'] = infos['total_file'] \
+            if infos.get('total_file') else 1
+        print '\n  ++ 正在下载: #', \
+            s % (1, 97, infos['nn']), \
+            '/', s % (1, 97, infos['total_file']), \
+            '#', col
 
         if args.aria2c:
             # 115 普通用户只能有4下载通道。
-            tlimit = ' --max-download-limit %s' % args.limit if args.limit else ''
+            tlimit = ' --max-download-limit %s' \
+                % args.limit if args.limit else ''
             cmd = 'aria2c -c -s4 -x4%s ' \
                 '-o "%s.tmp" -d "%s" ' \
                 '--user-agent "%s" ' \
@@ -262,7 +286,8 @@ def download(infos):
             cmd = 'wget -c%s ' \
                 '-O "%s.tmp" --user-agent "%s" ' \
                 '--header "Referer:http://m.115.com/" "%s"' \
-                % (tlimit, infos['file'], headers['User-Agent'], infos['dlink'])
+                % (tlimit, infos['file'], headers['User-Agent'],
+                   infos['dlink'])
 
         status = os.system(cmd)
         if status != 0:     # other http-errors, such as 302.
@@ -280,8 +305,12 @@ def play(infos):
         num = random.randint(0, 7) % 7
         col = s % (2, num + 90, infos['name'])
         infos['nn'] = infos['nn'] if infos.get('nn') else 1
-        infos['total_file'] = infos['total_file'] if infos.get('total_file') else 1
-        print '\n  ++ play: #', s % (1, 97, infos['nn']), '/', s % (1, 97, infos['total_file']), '#', col
+        infos['total_file'] = infos['total_file'] \
+            if infos.get('total_file') else 1
+        print '\n  ++ play: #', \
+            s % (1, 97, infos['nn']), '/', \
+            s % (1, 97, infos['total_file']), \
+            '#', col
 
         if not infos['purl']:
             print s % (1, 91, '  |-- m3u8 is not ready, using dlink')
@@ -300,9 +329,11 @@ def play(infos):
         else:
             pass
 
+    # TODO
     def exists(self, filepath):
         pass
 
+    # TODO
     def upload(self, path, dir_):
         pass
 
@@ -363,7 +394,8 @@ def do(self, pc):
             'file': t,
             'dir_': os.path.split(t)[0],
             'dlink': dlink,
-            #'purl': self._get_play_purl(pc) if args.play and self.is_vip else None,
+            #'purl': self._get_play_purl(pc) \
+            #   if args.play and self.is_vip else None,
             'purl': self._get_play_purl(pc) if args.play else None,
             'name': name,
             'nn': 1,
@@ -377,7 +409,8 @@ def main(argv):
 
     ######################################################
     # for argparse
-    p = argparse.ArgumentParser(description='download from 115.com reversely')
+    p = argparse.ArgumentParser(
+        description='download from 115.com reversely')
     p.add_argument('xxx', type=str, nargs='*', \
         help='命令对象.')
     p.add_argument('-a', '--aria2c', action='store_true', \
@@ -409,7 +442,8 @@ def main(argv):
             account = xxx[0]
             password = xxx[1]
         else:
-            print s % (1, 91, '  login\n  login account\n  login account password')
+            print s % (1, 91, '  login\n  login account\n  \
+                                 login account password')
 
         x = pan115()
         x.login(account, password)

91porn.py

@@ -67,8 +67,10 @@ def get_infos(self):
 
                 r = ss.get(apiurl, params=params)
                 if r.ok:
-                    dlink = re.search(r'file=(http.+?\.mp4)', r.content).group(1)
-                    name = re.search(r'viewkey=([\d\w]+)', self.url).group(1)
+                    dlink = re.search(
+                        r'file=(http.+?\.mp4)', r.content).group(1)
+                    name = re.search(
+                        r'viewkey=([\d\w]+)', self.url).group(1)
                     infos = {
                         'name': '%s.mp4' % name,
                         'file': os.path.join(os.getcwd(), '%s.mp4' % name),
@@ -86,7 +88,8 @@ def download(self, infos):
         col = s % (2, num + 90, infos['file'])
         print '\n  ++ 正在下载: %s' % col
 
-        cookies = '; '.join(['%s=%s' % (i, ii) for i, ii in ss.cookies.items()])
+        cookies = '; '.join(
+            ['%s=%s' % (i, ii) for i, ii in ss.cookies.items()])
         if args.aria2c:
             cmd = 'aria2c -c -x10 -s10 ' \
                 '-o "%s.tmp" -d "%s" --header "User-Agent: %s" ' \
@@ -118,7 +121,7 @@ def play(self, infos):
             '--http-header-fields "user-agent:%s" "%s"' \
             % (headers['User-Agent'], infos['dlink'])
 
-        status = os.system(cmd)
+        os.system(cmd)
         timeout = 1
         ii, _, _ = select.select([sys.stdin], [], [], timeout)
         if ii:
@@ -134,7 +137,8 @@ def main(url):
     x.do()
 
 if __name__ == '__main__':
-    p = argparse.ArgumentParser(description='download from 91porn.com')
+    p = argparse.ArgumentParser(
+        description='download from 91porn.com')
     p.add_argument('url', help='url of 91porn.com')
     p.add_argument('-a', '--aria2c', action='store_true', \
                 help='download with aria2c')

README.md

@@ -343,6 +343,7 @@
         -f number, --from_ number           从第几个开始(用于download, play)，eg: p /video -f 42
         -t ext, --type_ ext                 类型参数, 用 “,” 分隔
                                             eg:
+                                            p -t m3     # 播放流媒体(m3u8)
                                             s -t c      # 连续转存 (如果转存出错，再次运行命令
                                                         # 可以从出错的地方开始，用于转存大量文件时)
                                             l -t f      # 文件
@@ -427,6 +428,11 @@
         bp p http://pan.baidu.com/s/xxxxxxxxx -s [secret]
         bp p /movie -R     # 递归播放 /movie 中所有媒体文件
 
+        # 播放流媒体(m3u8)
+        上面的命令后加 -t m3
+        清晰度与在浏览器上播放的一样.
+        如果源文件是高清的(720P,1280P),那么流媒体会自动转为480P.
+
     离线下载:
 
         bp a http://mirrors.kernel.org/archlinux/iso/latest/archlinux-2014.06.01-dual.iso /path/to/save

bt.py

@@ -6,7 +6,6 @@
 import sys
 import re
 from hashlib import sha1
-import binascii
 import base64
 import requests
 import urlparse
@@ -15,18 +14,22 @@
 ############################################################
 headers = {
     "Connection": "keep-alive",
-    "Accept":"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
+    "Accept":"text/html,application/xhtml+xml,\
+        application/xml;q=0.9,image/webp,*/*;q=0.8",
     "Accept-Encoding":"gzip,deflate,sdch",
-    "Accept-Language":"en-US,en;q=0.8,zh-CN;q=0.6,zh;q=0.4,zh-TW;q=0.2",
-    "User-Agent":"Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.36 "\
-        "(KHTML, like Gecko) Chrome/32.0.1700.77 Safari/537.36"
+    "Accept-Language":"en-US,en;q=0.8,zh-CN;\
+        q=0.6,zh;q=0.4,zh-TW;q=0.2",
+    "User-Agent":"Mozilla/5.0 (X11; Linux x86_64) \
+        AppleWebKit/537.36 (KHTML, like Gecko) \
+        Chrome/40.0.2214.91 Safari/537.36"
 }
 
 ss = requests.session()
 ss.headers.update(headers)
 
 s = u'\x1b[%d;%dm%s\x1b[0m'       # terminual color template
-letters = [i for i in '.abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ' + '0123456789']
+letters = [i for i in '.abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ' \
+           + '0123456789']
 
 class bt(object):
     def transfer(self, string, tpath, foo=None, bar=None):
@@ -42,16 +45,18 @@ def transfer(self, string, tpath, foo=None, bar=None):
             for fl in dstring['info']['files']:
                 filename = fl['path'][-1]
                 if args.type_ == 'n':
-                    newfilename = re.sub(foo, bar, filename, re.I) if foo and bar else filename
+                    newfilename = re.sub(foo, bar, filename, re.I) \
+                        if foo and bar else filename
                     if filename != newfilename:
                         print filename, s % (1, 92, '==>'), newfilename
-                        path = [self._get_sub_dir_index(i) for i in fl['path'][:-1]] \
-                            + [newfilename]
+                        path = [self._get_sub_dir_index(i) \
+                                for i in fl['path'][:-1]]  + [newfilename]
                     else:
                         ext = os.path.splitext(filename)[-1]
                         ext = self._check_ext(ext)
-                        path = [self._get_sub_dir_index(i) for i in fl['path'][:-1]] \
-                            + ['%s%s' % (file_index, ext)]
+                        path = [self._get_sub_dir_index(i) \
+                                for i in fl['path'][:-1]] \
+                                + ['%s%s' % (file_index, ext)]
                     file_index += 1
                     fl['path'] = path
 
@@ -125,7 +130,8 @@ def do(url, proxies=None, data=None, timeout=None):
 
         ## with xunlei
         print s % (1, 94, '  >> try:'), 'bt.box.n0808.com'
-        url = 'http://bt.box.n0808.com/%s/%s/%s.torrent' % (hh[:2], hh[-2:], hh)
+        url = 'http://bt.box.n0808.com/%s/%s/%s.torrent' \
+            % (hh[:2], hh[-2:], hh)
         ss.headers['Referer'] = 'http://bt.box.n0808.com'
         result = do(url)
         if result: return result
@@ -207,7 +213,8 @@ def magnet2torrent(self, urls, dir_):
             string = self.get_torrent(hh)
             if string:
                 tpath = os.path.join(dir_, hh + '.torrent')
-                print s % (1, 97, '  ++ magnet to torrent:'), 'magnet:?xt=urn:btih:%s' % hh
+                print s % (1, 97, '  ++ magnet to torrent:'), \
+                    'magnet:?xt=urn:btih:%s' % hh
                 with open(tpath, 'w') as g:
                     g.write(string)
             else:
@@ -237,7 +244,8 @@ def trans(tpath):
                     tpath = path
                     trans(tpath)
             else:
-                print s % (1, 91, '  !! file doesn\'t existed'), s % (1, 93, '--'), path
+                print s % (1, 91, '  !! file doesn\'t existed'), \
+                    s % (1, 93, '--'), path
 
     def change(self, ups, dir_, foo=None, bar=None):
         for up in ups:
@@ -251,7 +259,8 @@ def change(self, ups, dir_, foo=None, bar=None):
                 string = self.get_torrent(hh)
                 if string:
                     tpath = os.path.join(dir_, hh + '.torrent')
-                    print s % (1, 97, '  ++ transfer:'), 'magnet:?xt=urn:btih:%s' % hh
+                    print s % (1, 97, '  ++ transfer:'), \
+                        'magnet:?xt=urn:btih:%s' % hh
                     self.transfer(string, tpath, foo=foo, bar=bar)
                 else:
                     print s % (1, 91, '  !! Can\'t get torrent from web.'), path
@@ -266,7 +275,8 @@ def do():
                                     print s % (1, 97, '  ++ transfer:'), ipath
                                     string = open(ipath).read()
                                     tpath = os.path.join(dir_, 'change_' + i)
-                                    self.transfer(string, tpath, foo=foo, bar=bar)
+                                    self.transfer(string, tpath, foo=foo,
+                                                  bar=bar)
                                     paths.update(ipath)
                                 if os.getcwd() == os.path.abspath(dir_):
                                     do()
@@ -277,10 +287,12 @@ def do():
                     if path.lower().endswith('torrent'):
                         print s % (1, 97, '  ++ transfer:'), path
                         string = open(path).read()
-                        tpath = os.path.join(dir_, 'change_' + os.path.basename(path))
+                        tpath = os.path.join(dir_,
+                                             'change_' + os.path.basename(path))
                         self.transfer(string, tpath, foo=foo, bar=bar)
             else:
-                print s % (1, 91, '  !! file doesn\'t existed'), s % (1, 93, '--'), path
+                print s % (1, 91, '  !! file doesn\'t existed'), \
+                    s % (1, 93, '--'), path
 
 def import_magnet(froms):
     ml = []
@@ -317,7 +329,8 @@ def get_magnet(cm):
 def main(argv):
     ######################################################
     # for argparse
-    p = argparse.ArgumentParser(description='magnet torrent 互转，数字命名bt内容文件名' \
+    p = argparse.ArgumentParser(
+        description='magnet torrent 互转，数字命名bt内容文件名' \
         ' 用法见 https://github.com/PeterDing/iScript')
     p.add_argument('xxx', type=str, nargs='*', \
         help='命令对象.')
@@ -337,11 +350,13 @@ def main(argv):
     comd = argv[1]
     xxx = args.xxx
 
-    dir_ = os.getcwd() if not args.directory else args.directory
+    dir_ = os.getcwd() if not args.directory \
+                        else args.directory
     if not os.path.exists(dir_):
         os.mkdir(dir_)
     if comd == 'm' or comd == 'mt':   # magnet to torrent
-        urls = xxx if not args.import_from else import_magnet(args.import_from)
+        urls = xxx if not args.import_from \
+                    else import_magnet(args.import_from)
         x = bt()
         x.magnet2torrent(urls, dir_)
 
@@ -351,14 +366,16 @@ def main(argv):
         x.torrent2magnet(paths)
 
     elif comd == 'c' or comd == 'ct':   # change
-        ups = xxx if not args.import_from else import_magnet(args.import_from)
+        ups = xxx if not args.import_from \
+                    else import_magnet(args.import_from)
         x = bt()
         x.change(ups, dir_, foo=None, bar=None)
 
     elif comd == 'cr' or comd == 'ctre':   # change
         foo = xxx[0]
         bar = xxx[1]
-        ups = xxx[2:] if not args.import_from else import_magnet(args.import_from)
+        ups = xxx[2:] if not args.import_from \
+                        else import_magnet(args.import_from)
         x = bt()
         x.change(ups, dir_, foo=foo, bar=bar)