misskey-microblogger/generate.py

30 lines
No EOL
685 B
Python

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()