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")
|
||||
|
||||
user_id = cur.lastrowid
|
||||
print(user_id)
|
||||
await cnx.commit()
|
||||
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")
|
||||
|
||||
blog_id = cur.lastrowid
|
||||
print(blog_id)
|
||||
await cnx.commit()
|
||||
return blog_id
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
dotenv
|
||||
fastapi
|
||||
uvicorn[standard]
|
||||
mysql-connector-python
|
||||
mysql-connector-python
|
||||
faker
|
||||
Loading…
Reference in New Issue