# coding=utf-8
from MyInstagram.models import User
from MyInstagram.models import City
import random


def gen_username():
    s = ""
    for i in range(random.randint(8, 12)):
        s += random.choice("qwertyuiopasdfghjklzcvbxnm1234567890")
    return s


def gen_email():
    s = ""
    for i in range(random.randint(13, 20)):
        s += random.choice("qwertyuiopasdfghjklzcvbxnm1234567890")
    s += random.choice(["@gmail.com", "@mail.ru", "@yandex.ru", "@rambler.ru"])
    return s


def gen_password():
    s = ""
    for i in range(random.randint(20, 30)):
        s += random.choice("qwertyuiopasdfghjklzcvbxnm1234567890*.@#$%")
    return s


def gen_first_name(sex=0):
    if sex == 0:
        f = open('Generate/names-m.txt')
    else:
        f = open('Generate/names-f.txt')
    names = f.readlines()
    name = random.choice(names).strip()
    f.close()
    return name


def get_last_name(sex=0):
    f = open('Generate/surnames.txt')
    surnames = f.readlines()
    surname = random.choice(surnames).strip()
    if sex == 1:
        surname += u"а".encode("utf-8")
    f.close()
    return surname


def get_random_city():
    f = open('Generate/cities.txt')
    cities = f.readlines()
    city = random.choice(cities).strip()
    obj, created = City.objects.get_or_create(city=city)
    return obj


def gen_user():
    username = gen_username()
    email = gen_email()
    password = gen_password()
    usr = User(username=username, email=email, password=password)
    if random.randint(0, 30) <= 20:
        if random.randint(0, 1) == 0:
            usr.first_name = gen_first_name()
            usr.last_name = get_last_name()
        else:
            usr.first_name = gen_first_name(sex=1)
            usr.last_name = get_last_name(sex=1)
    if random.randint(0, 20) <= 15:
        usr.city = get_random_city()
    return usr


def gen_users(n=10000):
    users = []
    for i in range(n):
        users.append(gen_user())
    User.objects.bulk_create(users)