faker dummy inserter 추가, mysql 모듈 debug print 제거
parent
b7f1588bc8
commit
a951d0d1b6
|
|
@ -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())
|
||||||
|
|
@ -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
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
dotenv
|
dotenv
|
||||||
fastapi
|
fastapi
|
||||||
uvicorn[standard]
|
uvicorn[standard]
|
||||||
mysql-connector-python
|
mysql-connector-python
|
||||||
|
faker
|
||||||
Loading…
Reference in New Issue