import sys
import json
import typing

from src.User.User import User
from src.Generator.GenerateUser import GenerateUser
from src.Generator.GenerateRelationships import GenerateRelationships

DEFAULT_GENERATE_USER_COUNT = 5

def generate_user() -> User:
	while True:
		user = GenerateUser()
		print(f"1. Create username: {user.username}")
		#user.set_api_key(input("2. Paste API key:"))

		yield user.autorun()

def main():
	users = []

	for i in (range(sys.argv[1] if len(sys.argv) > 1 else DEFAULT_GENERATE_USER_COUNT)):
		print(f"\nGenerating user: {i + 1}")
		users.append(next(generate_user()))
		i += 1

	GenerateRelationships(users).autorun()

if __name__ == "__main__":
	main()