_ Git - youtube-dl/blob - devscripts/youtube_genalgo.py

   1 #!/usr/bin/env python
   2
   3 # Generate youtube signature algorithm from test cases
   4
   5 import sys
   6
   7 tests = [
   8     # 92 - vflQw-fB4 2013/07/17
   9     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~\"",
  10      "mrtyuioplkjhgfdsazxcvbnq1234567890QWERTY}IOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]\"|:;"),
  11     # 90
  12     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`",
  13      "mrtyuioplkjhgfdsazxcvbne1234567890QWER[YUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={`]}|"),
  14     # 88
  15     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<",
  16      "J:|}][{=+-_)(*&;%$#@>MNBVCXZASDFGH^KLPOIUYTREWQ0987654321mnbvcxzasdfghrklpoiuytej"),
  17     # 87 - vflART1Nf 2013/07/24
  18     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$^&*()_-+={[]}|:;?/>.<",
  19      "tyuioplkjhgfdsazxcv<nm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$^&*()_-+={[]}|:;?/>"),
  20     # 86 - vfl_ymO4Z 2013/06/27
  21     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[|};?/>.<",
  22      "ertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!/#$%^&*()_-+={[|};?@"),
  23     # 85 - vflSAFCP9 2013/07/19
  24     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[};?/>.<",
  25      "ertyuiqplkjhgfdsazx$vbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#<%^&*()_-+={[};?/c"),
  26     # 84
  27     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[};?>.<",
  28      "<.>?;}[{=+-_)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYTREWe098765432rmnbvcxzasdfghjklpoiuyt1"),
  29     # 83 - vflcaqGO8 2013/07/11
  30     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!#$%^&*()_+={[};?/>.<",
  31      "urty8ioplkjhgfdsazxcvbqm1234567S90QWERTYUIOPLKJHGFDnAZXCVBNM!#$%^&*()_+={[};?/>.<"),
  32     # 82
  33     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.<",
  34      "Q>/?;}[{=+-(*<^%$#@!MNBVCXZASDFGHKLPOIUY8REWT0q&7654321mnbvcxzasdfghjklpoiuytrew9"),
  35     # 81 - vflLC8JvQ 2013/07/25
  36     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.",
  37      "C>/?;}[{=+-(*&^%$#@!MNBVYXZASDFGHKLPOIU.TREWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  38     # 79 - vflLC8JvQ 2013/07/25 (sporadic)
  39     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/",
  40      "Z?;}[{=+-(*&^%$#@!MNBVCXRASDFGHKLPOIUYT/EWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  41 ]
  42
  43 def find_matching(wrong, right):
  44     idxs = [wrong.index(c) for c in right]
  45     return compress(idxs)
  46     return ('s[%d]' % i for i in idxs)
  47
  48 def compress(idxs):
  49     def _genslice(start, end, step):
  50         starts = '' if start == 0 else str(start)
  51         ends = ':%d' % (end+step)
  52         steps = '' if step == 1 else (':%d' % step)
  53         return 's[%s%s%s]' % (starts, ends, steps)
  54
  55     step = None
  56     for i, prev in zip(idxs[1:], idxs[:-1]):
  57         if step is not None:
  58             if i - prev == step:
  59                 continue
  60             yield _genslice(start, prev, step)
  61             step = None
  62             continue
  63         if i - prev in [-1, 1]:
  64             step = i - prev
  65             start = prev
  66             continue
  67         else:
  68             yield 's[%d]' % prev
  69     if step is None:
  70         yield 's[%d]' % i
  71     else:
  72         yield _genslice(start, i, step)
  73
  74 def _assert_compress(inp, exp):
  75     res = list(compress(inp))
  76     if res != exp:
  77         print('Got %r, expected %r' % (res, exp))
  78         assert res == exp
  79 _assert_compress([0,2,4,6], ['s[0]', 's[2]', 's[4]', 's[6]'])
  80 _assert_compress([0,1,2,4,6,7], ['s[:3]', 's[4]', 's[6:8]'])
  81 _assert_compress([8,0,1,2,4,7,6,9], ['s[8]', 's[:3]', 's[4]', 's[7:5:-1]', 's[9]'])
  82
  83 def gen(wrong, right, indent):
  84     code = ' + '.join(find_matching(wrong, right))
  85     return 'if len(s) == %d:\n%s    return %s\n' % (len(wrong), indent, code)
  86
  87 def genall(tests):
  88     indent = ' ' * 8
  89     return indent + (indent + 'el').join(gen(wrong, right, indent) for wrong,right in tests)
  90
  91 def main():
  92     print(genall(tests))
  93
  94 if __name__ == '__main__':
  95     main()