import requests
import os
import random
import csv
from colorama import init, Fore, Back, Style

headers = {
    "accept": "application/json, text/plain, */*",
    "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.131 Safari/537.36"
}

#Нужен для colorama
init(autoreset=True)



#Генератор
def generate():
    sim = "qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM123456789"
    col = int(input("Введите кол-во: "))
    leing = int(input("Введите длину:"))
    spisok = []



    for i in range(int(col)):
        t = ''
        for a in range(int(leing)):
            t += random.choice(sim)
            url = f"https://www.tiktok.com/@{t}"
            r = requests.get(url=url, headers=headers)       
        if r.status_code == 200:
            spisok.append(url)
            print(Fore.GREEN + '[200]' + url)
        else:
            print(Fore.RED + '[404]' + url)

    #Запись в CSV
    for users in spisok:
        with open("data.csv", "a") as file:
            writer = csv.writer(file)
            writer.writerow(([users]))

    poftor = input("Повторить? y/n: ")
    if poftor == 'y':
        generate()
    else:
        pass

generate()