import random
import string
import argparse
import subprocess
import MySQLdb

def save_to_db(addr_arr):
    check_proc = subprocess.Popen('ls', stdout=subprocess.PIPE)
    tmp = check_proc.stdout.read()
    if 'credentials.txt' in tmp:
        credentials = open('credentials.txt','r')
        credentials = credentials.read()
        credentials = credentials.split(':')
        hostname = credentials[0]
        username = credentials[1]
        password = credentials[2]
        dbname = credentials[3]
        database = MySQLdb.connect(hostname, username, password, dbname)
        print("AUTOCONNECT OK")
    else:
        print("hostname: ")
        hostname = raw_input('>')
        print("username: ")
        username = raw_input('>')
        print("password: ")
        password = raw_input('>')
        print("dbname: ")
        dbname = raw_input('>')
        print("Do you want to remember this data? [y/n]")
        save = raw_input('>')
        if save == 'y':
            credentials = open('credentials.txt','a')
            credentials.write(str(hostname) + ":" + str(username) + ":" + str(password) + ":" + str(dbname))
        database = MySQLdb.connect(hostname, username, password, dbname)
        print("OK")

    print("Query: (INSERT INTO ____ (____) )")
    userquery = input('>')
    for i in range(len(addr_arr)):
        DBquery = (str(userquery) + "VALUES " + str(url_list[i]))
    cursor = db.cursor()


def write_to_file(addr_arr):
    urllist = open('urllist.md', 'a')
    for addr in range(len(addr_arr)):
        urllist.write(str(addr_arr[addr]) + "\n")

def check_unique(rnd_url):
    # Checks if directory of given name exists
    check_proc = subprocess.Popen('ls', stdout=subprocess.PIPE)
    tmp = check_proc.stdout.read()
    tof_arr = []
    for i in rnd_url:
        tof_arr.append(i in tmp)
    retval = False in tof_arr
    return retval

def rand_url_gen(leng, rounds):
    rnd_url_list = []
    for i in range(rounds):
        rnd_url = ''.join([random.choice(string.ascii_letters + string.digits) for n in xrange(leng)])
        rnd_url_list.append(rnd_url)
    return rnd_url_list

def makedir(url_name, rounds):
    print("Generated addresses: " + str(url_name))
    for i in range(rounds):
        subprocess.call(['mkdir', url_name[i]])

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("num", help = "the length of desired url", type = int)
    parser.add_argument("amount", help = "amount of desired addresses", type = int)
    parser.add_argument("--dbs", help = "append addresses to database", dest="dbs", action="store_true")
    parser.add_argument("--nd", help = "do not generate directories", dest="nd", action="store_true")
    parser.set_defaults(dbs=False,help=True)
    args = parser.parse_args()

    url_list = rand_url_gen(args.num, args.amount)

    if check_unique(url_list) == True and args.nd:
        makedir(url_list, args.amount)
    else:
        rand_url_gen(args.num, args.amount)

    write_to_file(url_list)

    if args.dbs:
        # print("DB OK")
        save_to_db(url_list)

if __name__ == "__main__":
    main()