diff options
Diffstat (limited to 'main.py')
| -rwxr-xr-x | main.py | 65 |
1 files changed, 64 insertions, 1 deletions
@@ -1,6 +1,14 @@ #!/usr/bin/env python3 -import argparse, sys, json, os, re +import argparse +import sys +import json +import os +import re +import time +import datetime +from pprint import pprint + try: from termcolor import cprint colors_supported = True @@ -190,6 +198,9 @@ def decode3(s): return buf + + + # s: source # t: type def decode_auto(s, t, reverse_decoded=False, remove_junk=True): @@ -209,6 +220,10 @@ def decode_auto(s, t, reverse_decoded=False, remove_junk=True): return result + +def sort_data_by_date(item): + return int(time.mktime(datetime.datetime.strptime(item['date'], '%d/%m/%y').timetuple())) + def main(): parser = argparse.ArgumentParser() @@ -220,6 +235,7 @@ def main(): parser.add_argument('--is-url', action='store_true') parser.add_argument('--type', type=int, choices=[1, 2, 3], default=1) parser.add_argument('--reverse-decoded', action='store_true') + parser.add_argument('--gen-page', action='store_true') args = parser.parse_args() data = load_data() @@ -273,5 +289,52 @@ def main(): count = len(data) print("Total texts: %s" % count) + elif args.gen_page: + # sort by date + data = sorted(data, key=sort_data_by_date) + #pprint(data) + + buf = [] + for post in data: + cipher_type = post['type'] if 'type' in post else 1 + decoded_text = decode_auto(post['text'], cipher_type) + + post_buf = '' + post_buf += '**Дата**: %s\n' % post['date'] + + if 'pic' in post and post['pic']: + # make sure it is a list + pic = post['pic'] if isinstance(post['pic'], list) else [post['pic']] + + pic_buf = [] + for p in pic: + pic_buf.append('' % p) + post_buf += '**Пикрилейтед:** %s\n' % ''.join(pic_buf) + + if 'link' in post: + # make sure it is a list + link = post['link'] if isinstance(post['link'], list) else [post['link']] + + link_buf = [] + for l in link: + link_buf.append('[%s](%s)' % (l, l)) + post_buf += '**Ссылки:** %s\n' % ', '.join(link_buf) + + if 'source' in post: + post_buf += '**Источник:** %s' % post['source'] + if 'source_link' in post: + post_buf += ', [%s](%s)\n' % (post['source_link'], post['source_link']) + else: + post_buf += '\n' + + post_buf += '**Шифровка (тип %d)**: \n```\n%s\n```\n' % (cipher_type, post['text']) + post_buf += '**Расшифровка:** \n```\n%s\n```' % decoded_text + + buf.append(post_buf) + + md = "\n---\n".join(buf) + with open(os.path.join(CWD, 'data.md'), 'w') as f: + f.write(md) + if __name__ == '__main__': sys.exit(main()) |