devscripts/youtube_genalgo.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3
   4 # Generate youtube signature algorithm from test cases
   5
   6 import sys
   7
   8 tests = [
   9     # 93 - vfl79wBKW 2013/07/20
  10     (u"qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~\"€",
  11      u".>/?;:|}][{=+-_)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYTREWQ098765'321mnbvcxzasdfghjklpoiu"),
  12     # 92 - vflQw-fB4 2013/07/17
  13     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~\"",
  14      "mrtyuioplkjhgfdsazxcvbnq1234567890QWERTY}IOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]\"|:;"),
  15     # 91 - vfl79wBKW 2013/07/20 (sporadic)
  16     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~",
  17      "/?;:|}][{=+-_)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYTREWQ09876543.1mnbvcxzasdfghjklpoiu"),
  18     # 90
  19     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`",
  20      "mrtyuioplkjhgfdsazxcvbne1234567890QWER[YUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={`]}|"),
  21     # 89
  22     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'",
  23      "/?;:|}<[{=+-_)(*&^%$#@!MqBVCXZASDFGHJKLPOIUYTREWQ0987654321mnbvcxzasdfghjklpoiuyt"),
  24     # 88 - vflapUV9V 2013/08/28
  25     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<",
  26      "ioplkjhgfdsazxcvbnm12<4567890QWERTYUIOZLKJHGFDSAeXCVBNM!@#$%^&*()_-+={[]}|:;?/>.3"),
  27     # 87
  28     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$^&*()_-+={[]}|:;?/>.<",
  29      "uioplkjhgfdsazxcvbnm1t34567890QWE2TYUIOPLKJHGFDSAZXCVeNM!@#$^&*()_-+={[]}|:;?/>.<"),
  30     # 86 - vflHql6Pr 2013/09/24
  31     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[|};?/>.<",
  32      ";}|[{=+-d)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYT_EWQ0987654321mnbvcxzas/fghjklpoiuytrewq"),
  33     # 85 - vflkuzxcs 2013/09/11
  34     ('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"#$%&\'()*+,-./:;<=>?@[',
  35      '3456789a0cdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS[UVWXYZ!"#$%&\'()*+,-./:;<=>?@'),
  36     # 84 - vflHql6Pr 2013/09/24 (sporadic)
  37     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[};?>.<",
  38      "}[{=+-_)g*&^%$#@!MNBVCXZASDFGHJKLPOIUYTRE(Q0987654321mnbvcxzasdf?hjklpoiuytrewq"),
  39     # 83
  40     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!#$%^&*()_+={[};?/>.<",
  41      ".>/?;}[{=+_)(*&^%<#!MNBVCXZASPFGHJKLwOIUYTREWQ0987654321mnbvcxzasdfghjklpoiuytreq"),
  42     # 82 - vflGNjMhJ 2013/09/12
  43     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.<",
  44      ".>/?;}[<=+-(*&^%$#@!MNBVCXeASDFGHKLPOqUYTREWQ0987654321mnbvcxzasdfghjklpoiuytrIwZ"),
  45     # 81 - vflLC8JvQ 2013/07/25
  46     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.",
  47      "C>/?;}[{=+-(*&^%$#@!MNBVYXZASDFGHKLPOIU.TREWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  48     # 80 - vflZK4ZYR 2013/08/23 (sporadic)
  49     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>",
  50      "wertyuioplkjhgfdsaqxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&z(-+={[};?/>"),
  51     # 79 - vflLC8JvQ 2013/07/25 (sporadic)
  52     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/",
  53      "Z?;}[{=+-(*&^%$#@!MNBVCXRASDFGHKLPOIUYT/EWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  54 ]
  55
  56 tests_age_gate = [
  57     # 86 - vflqinMWD
  58     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[|};?/>.<",
  59      "ertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!/#$%^&*()_-+={[|};?@"),
  60 ]
  61
  62 def find_matching(wrong, right):
  63     idxs = [wrong.index(c) for c in right]
  64     return compress(idxs)
  65     return ('s[%d]' % i for i in idxs)
  66
  67 def compress(idxs):
  68     def _genslice(start, end, step):
  69         starts = '' if start == 0 else str(start)
  70         ends = ':%d' % (end+step)
  71         steps = '' if step == 1 else (':%d' % step)
  72         return 's[%s%s%s]' % (starts, ends, steps)
  73
  74     step = None
  75     for i, prev in zip(idxs[1:], idxs[:-1]):
  76         if step is not None:
  77             if i - prev == step:
  78                 continue
  79             yield _genslice(start, prev, step)
  80             step = None
  81             continue
  82         if i - prev in [-1, 1]:
  83             step = i - prev
  84             start = prev
  85             continue
  86         else:
  87             yield 's[%d]' % prev
  88     if step is None:
  89         yield 's[%d]' % i
  90     else:
  91         yield _genslice(start, i, step)
  92
  93 def _assert_compress(inp, exp):
  94     res = list(compress(inp))
  95     if res != exp:
  96         print('Got %r, expected %r' % (res, exp))
  97         assert res == exp
  98 _assert_compress([0,2,4,6], ['s[0]', 's[2]', 's[4]', 's[6]'])
  99 _assert_compress([0,1,2,4,6,7], ['s[:3]', 's[4]', 's[6:8]'])
 100 _assert_compress([8,0,1,2,4,7,6,9], ['s[8]', 's[:3]', 's[4]', 's[7:5:-1]', 's[9]'])
 101
 102 def gen(wrong, right, indent):
 103     code = ' + '.join(find_matching(wrong, right))
 104     return 'if len(s) == %d:\n%s    return %s\n' % (len(wrong), indent, code)
 105
 106 def genall(tests):
 107     indent = ' ' * 8
 108     return indent + (indent + 'el').join(gen(wrong, right, indent) for wrong,right in tests)
 109
 110 def main():
 111     print(genall(tests))
 112     print(u'    Age gate:')
 113     print(genall(tests_age_gate))
 114
 115 if __name__ == '__main__':
 116     main()