faker dummy inserter 추가, mysql 모듈 debug print 제거

main
jaehwang 2025-11-10 10:56:41 +09:00
parent b7f1588bc8
commit a951d0d1b6
3 changed files with 33 additions and 3 deletions

View File

@ -0,0 +1,31 @@
import asyncio
from faker import Faker
import module.mysql_utils as sql
from module.pydantic_models import *
fake = Faker('ko_KR')
unique_num = 1
INSERT_NUM = 1000
class UserFactory:
unique_num = 0
@classmethod
def generate_user_name(cls):
cls.unique_num += 1
return f"{fake.name()}{cls.unique_num:04d}"
@classmethod
def generate_phone_number(cls):
return fake.phone_number()
@classmethod
def create(cls):
return (cls.generate_user_name(), cls.generate_phone_number())
async def main():
for i in range(INSERT_NUM):
await sql.create_user(*UserFactory.create())
# await asyncio.gather(*[sql.create_user(*UserFactory.create()) for i in range(INSERT_NUM)]) # 커넥션 INSERT NUM개수만큼 생기는 문제 Semaphore 등으로 해결해야 쓸수있음
return
if __name__ == "__main__":
asyncio.run(main())

View File

@ -34,7 +34,6 @@ async def create_user(user_name:str, phone_number:str=None):
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="user already exist") raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="user already exist")
user_id = cur.lastrowid user_id = cur.lastrowid
print(user_id)
await cnx.commit() await cnx.commit()
return user_id return user_id
@ -148,7 +147,6 @@ async def create_blog(blog_owner:int, blog_title:str, blog_content:str = None):
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="blog already exist") raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="blog already exist")
blog_id = cur.lastrowid blog_id = cur.lastrowid
print(blog_id)
await cnx.commit() await cnx.commit()
return blog_id return blog_id

View File

@ -2,3 +2,4 @@ dotenv
fastapi fastapi
uvicorn[standard] uvicorn[standard]
mysql-connector-python mysql-connector-python
faker