20 lines
414 B
Python
20 lines
414 B
Python
|
import asyncio
|
||
|
|
import asyncpg
|
||
|
|
import os
|
||
|
|
from asyncpg.connection import Connection
|
||
|
|
|
||
|
|
|
||
|
|
async def drop_tables(connection: Connection):
|
||
|
|
await connection.execute("DROP TABLE users, projects CASCADE")
|
||
|
|
|
||
|
|
|
||
|
|
async def main():
|
||
|
|
DB_URL = os.getenv("DB_URL")
|
||
|
|
connection = await asyncpg.connect(DB_URL)
|
||
|
|
await drop_tables(connection)
|
||
|
|
await connection.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
asyncio.run(main())
|