.

app.py

@@ -1,10 +1,12 @@
 from fastapi import FastAPI
-from routers import test_router
+from routers import test_router, testdb_router, user_router
 import uvicorn
 
 app = FastAPI()
 
 app.include_router(test_router.router)
+app.include_router(testdb_router.router)
+app.include_router(user_router.router)
 
 
 @app.get("/")
@@ -18,4 +20,4 @@ if __name__ == "__main__":
         host="127.0.0.1",
         port=8000,
         reload=True
-    )
+    )

envinfo.text

@@ -0,0 +1,2 @@
+POSTGRES_URL="postgres://postgres.hjgdlilnfccakdsmsypg:qBXsw226lYJabYgp@aws-1-ap-northeast-2.pooler.supabase.com:6543/postgres?sslmode=require"
+POSTGRES_URL_ORIGINAL="postgres://postgres.hjgdlilnfccakdsmsypg:qBXsw226lYJabYgp@aws-1-ap-northeast-2.pooler.supabase.com:6543/postgres?sslmode=require&supa=base-pooler.x"

routers/test_router.py

@@ -61,4 +61,4 @@ async def upload_file(file: UploadFile = File(...)):
         }
         return CommonResponse(success=True, data=data)
     except Exception as e:
-        return CommonResponse(success=False, msg=str(e))
+        return CommonResponse(success=False, msg=str(e))

routers/testdb_router.py

@@ -0,0 +1,106 @@
+from fastapi import APIRouter, Form, File, UploadFile
+from utils.common import CommonResponse
+from utils.db import get_connection
+
+router = APIRouter(
+    prefix="/testdb",
+    tags=["testdb"]
+)
+
+@router.get("/")
+def test_home():
+    try:
+        return CommonResponse(success=True)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))
+
+@router.get("/now")
+def select_now():
+    try:
+        with get_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute("select now()")
+                row = cur.fetchone()
+
+        data = {
+            "now": row[0].isoformat() if row else None
+        }
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))
+
+@router.get("/select_test")
+def select_now(id: int = 0):
+    try:
+        with get_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute("""
+                SELECT
+                *
+                FROM t_test
+                WHERE
+                    CASE
+                        WHEN %s != 0 THEN id = %s
+                        ELSE TRUE
+                    END
+                LIMIT 1000
+                """, [id, id])
+                row = cur.fetchall()
+
+        data = row
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))
+
+@router.post("/upsert_test")
+def upsert_test(id: int = Form(0), name: str = Form("")):
+    try:
+        with get_connection() as conn:
+            with conn.cursor() as cur:
+                if id <= 0:
+                    cur.execute("""
+                    INSERT INTO t_test (name)
+                    VALUES (%s)
+                    RETURNING id, name
+                    """, [name])
+                else:
+                    cur.execute("""
+                    UPDATE t_test
+                    SET name = %s
+                    WHERE id = %s
+                    RETURNING id, name
+                    """, [name, id])
+
+                row = cur.fetchone()
+                if not row:
+                    conn.rollback()
+                    return CommonResponse(success=False, msg="data not found")
+
+                conn.commit()
+
+        data = {
+            "id": row[0],
+            "name": row[1]
+        }
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))
+
+@router.post("/delete_test")
+def delete_test(id: int = Form(0)):
+    try:
+        with get_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute("""
+                DELETE FROM t_test
+                WHERE id = %s
+                """, [id])
+
+                conn.commit()
+
+        data = {
+            "msg" : "delete success"
+        }
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))

routers/user_router.py

@@ -0,0 +1,93 @@
+from fastapi import APIRouter, Form, File, UploadFile
+from utils.common import CommonResponse
+from utils.db import get_connection
+
+router = APIRouter(
+    prefix="/user",
+    tags=["user"]
+)
+
+@router.post("/login_register")
+def login_register(provider: str = Form(""), firebase_uid: str = Form(""),
+                   email: str = Form(""),display_name: str = Form(""),
+                   photo_url: str = Form("")):
+    try:
+        if firebase_uid == "":
+            return CommonResponse(success=False, msg="firebase_uid is required")
+        if email == "":
+            return CommonResponse(success=False, msg="email is required")
+
+        with get_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute("""
+                INSERT INTO t_user (
+                    firebase_uid,
+                    email,
+                    display_name,
+                    photo_url,
+                    provider,
+                    last_login_at
+                )
+                VALUES (
+                    %s,
+                    %s,
+                    NULLIF(%s, ''),
+                    NULLIF(%s, ''),
+                    COALESCE(NULLIF(%s, ''), 'google'),
+                    CURRENT_TIMESTAMP
+                )
+                ON CONFLICT (firebase_uid)
+                DO UPDATE SET
+                    email = EXCLUDED.email,
+                    display_name = EXCLUDED.display_name,
+                    photo_url = EXCLUDED.photo_url,
+                    provider = EXCLUDED.provider,
+                    last_login_at = CURRENT_TIMESTAMP,
+                    updated_at = CURRENT_TIMESTAMP
+                RETURNING
+                    id,
+                    firebase_uid,
+                    email,
+                    display_name,
+                    photo_url,
+                    provider,
+                    role,
+                    status,
+                    last_login_at,
+                    created_at,
+                    updated_at
+                """, [firebase_uid, email, display_name, photo_url, provider])
+
+                row = cur.fetchone()
+                conn.commit()
+
+        data = {
+            "id": row[0],
+            "firebase_uid": row[1],
+            "email": row[2],
+            "display_name": row[3],
+            "photo_url": row[4],
+            "provider": row[5],
+            "role": row[6],
+            "status": row[7],
+            "last_login_at": row[8].isoformat() if row[8] else None,
+            "created_at": row[9].isoformat() if row[9] else None,
+            "updated_at": row[10].isoformat() if row[10] else None
+        }
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))
+
+@router.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        # business logic
+        contents = await file.read()
+        data = {
+            "filename": file.filename,
+            "content_type": file.content_type,
+            "size": len(contents)
+        }
+        return CommonResponse(success=True, data=data)
+    except Exception as e:
+        return CommonResponse(success=False, msg=str(e))

utils/common.py

@@ -6,4 +6,5 @@ T = TypeVar("T")
 class CommonResponse(BaseModel, Generic[T]):
     success: bool=True
     data: Optional[T] = None
-    msg: str = ""
+    msg: str = ""
+

utils/db.py

@@ -0,0 +1,67 @@
+import os
+from urllib.parse import parse_qsl, urlencode, urlsplit, urlunsplit
+
+import psycopg
+from dotenv import load_dotenv
+
+load_dotenv()
+
+ALLOWED_POSTGRES_PARAMS = {
+    "application_name",
+    "connect_timeout",
+    "dbname",
+    "fallback_application_name",
+    "gssencmode",
+    "host",
+    "hostaddr",
+    "keepalives",
+    "keepalives_count",
+    "keepalives_idle",
+    "keepalives_interval",
+    "load_balance_hosts",
+    "options",
+    "passfile",
+    "password",
+    "port",
+    "replication",
+    "require_auth",
+    "requiressl",
+    "service",
+    "sslcert",
+    "sslcompression",
+    "sslcrl",
+    "sslcrldir",
+    "sslkey",
+    "sslmode",
+    "sslnegotiation",
+    "sslpassword",
+    "sslrootcert",
+    "sslsni",
+    "target_session_attrs",
+    "tcp_user_timeout",
+    "user",
+}
+
+
+def get_postgres_url():
+    postgres_url = os.getenv("POSTGRES_URL")
+    if not postgres_url:
+        raise ValueError("POSTGRES_URL is not set")
+
+    return clean_postgres_url(postgres_url)
+
+
+def clean_postgres_url(postgres_url):
+    parts = urlsplit(postgres_url.strip().strip("\"'"))
+    query = urlencode(
+        [
+            (key, value)
+            for key, value in parse_qsl(parts.query, keep_blank_values=True)
+            if key in ALLOWED_POSTGRES_PARAMS
+        ]
+    )
+    return urlunsplit((parts.scheme, parts.netloc, parts.path, query, parts.fragment))
+
+
+def get_connection():
+    return psycopg.connect(get_postgres_url())