#!/usr/bin/python

import sys
import datetime
import os
import libtorrent as lt

def main(argv):
    date = argv[1]
    dir = argv[2]
    try:
        datetime.datetime.strptime(date, '%d%m%Y')
    except:
        raise Exception('Wrong date format')
    try:
        os.path.isdir(dir)
    except:
        raise Exception('Wrong dir format')
    print(argv[2])
    createtorrent(date,dir)

def createtorrent(date,dir):
    fs = lt.file_storage()
    piece_size = 128 * 1024
    lt.add_files(fs, dir)
    t = lt.create_torrent(fs)
    t.set_comment("test")

    torrent = t.generate()
    f = open("Setup-Abrites-Common-{}.torrent".format(date), "wb")
    f.write(lt.bencode(torrent))
    f.close()


def help():
    print("Input date(DDMMYYYY) and directory")

if __name__ == "__main__":
    if len(sys.argv)==1:
        print("Please provide date and directory, for help see --help")
    elif sys.argv[1] == "--help":
        help()
    else:
        main(sys.argv)