Post table create

main.py

@@ -4,26 +4,33 @@ from sqlalchemy import create_engine, Column, Integer, String
 from sqlalchemy.orm import sessionmaker, declarative_base, Session
 import bcrypt
 from starlette.middleware.cors import CORSMiddleware
+from sqlalchemy import ForeignKey, DateTime, Text
+from sqlalchemy.orm import relationship
+from datetime import datetime
 
+import os, uuid
+from typing import Optional, List
+from fastapi import File, UploadFile, Form
+from starlette.staticfiles import StaticFiles
+from pydantic import BaseModel as PModel
 
 # ----------------- MySQL Connection -----------------
 DATABASE_URL = "postgresql+psycopg2://city_university_db_user:au84DXp5L55SYrir23DzrezulwqSJZzc@dpg-d2gitojuibrs73ed7s00-a.oregon-postgres.render.com:5432/city_university_db"
 engine = create_engine(DATABASE_URL, pool_pre_ping=True)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
-Base = declarative_base()
-
-
 # ----------------- ORM Model (matches user_accounts table) -----------------
-class UserDB(Base):
-    __tablename__ = "user_accounts"
-    id = Column(Integer, primary_key=True, index=True)
-    full_name = Column(String(100), nullable=False)
-    email = Column(String(100), unique=True, index=True, nullable=False)
-    password = Column(String(255), nullable=False)
 
 
 app = FastAPI()
 
+
+
+Base = declarative_base()
+UPLOAD_DIR = "uploads"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+app.mount("/uploads", StaticFiles(directory=UPLOAD_DIR), name="uploads")
+
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],          # allow all origins
@@ -32,6 +39,14 @@ app.add_middleware(
     allow_headers=["*"],          # Authorization, Content-Type, etc.
 )
 
+
+class UserDB(Base):
+    __tablename__ = "user_accounts"
+    id = Column(Integer, primary_key=True, index=True)
+    full_name = Column(String(100), nullable=False)
+    email = Column(String(100), unique=True, index=True, nullable=False)
+    password = Column(String(255), nullable=False)
+
 # Dependency for getting DB session in routes
 def get_db():
     db = SessionLocal()
@@ -54,6 +69,27 @@ class ResetPassword(BaseModel):
     email: EmailStr
     new_password: str
 
+class PostOut(PModel):
+    id: int
+    user_id: int
+    content: Optional[str] = None
+    image_url: Optional[str] = None
+    created_at: datetime
+    class Config:
+        from_attributes = True  # pydantic v2
+
+
+class PostDB(Base):
+    __tablename__ = "posts"
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(Integer, ForeignKey("user_accounts.id", ondelete="CASCADE"), nullable=False)
+    content = Column(Text, nullable=True)                 # text body (optional)
+    image_url = Column(String(300), nullable=True)        # store public URL or /uploads/<file>
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+
+    user = relationship("UserDB")  # optional convenience
+
+    
 
 @app.get("/")
 def home():
@@ -94,4 +130,42 @@ def forgot_password(reset: ResetPassword, db: Session = Depends(get_db)):
     hashed_pw = bcrypt.hashpw(reset.new_password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
     db_user.password = hashed_pw
     db.commit()
-    return {"message": "Password updated successfully"}
+    return {"message": "Password updated successfully"}
+
+@app.post("/posts", response_model=PostOut)
+async def create_post(
+    user_id: int = Form(...),
+    content: Optional[str] = Form(None),
+    image: Optional[UploadFile] = File(None),
+    db: Session = Depends(get_db),
+):
+    # require at least one of content or image
+    if (not content or not content.strip()) and image is None:
+        raise HTTPException(status_code=400, detail="Provide content or an image")
+
+    # verify user exists
+    user = db.get(UserDB, user_id)
+    if not user:
+        raise HTTPException(status_code=404, detail="User not found")
+
+    image_url = None
+    if image is not None:
+        allowed = {"image/png", "image/jpeg", "image/jpg", "image/webp"}
+        if image.content_type not in allowed:
+            raise HTTPException(status_code=415, detail="Unsupported image type")
+        ext = { "image/png": ".png", "image/jpeg": ".jpg", "image/jpg": ".jpg", "image/webp": ".webp" }[image.content_type]
+        filename = f"{uuid.uuid4().hex}{ext}"
+        path = os.path.join(UPLOAD_DIR, filename)
+        with open(path, "wb") as f:
+            f.write(await image.read())
+        image_url = f"/uploads/{filename}"  # public via static mount
+
+    post = PostDB(user_id=user_id, content=content, image_url=image_url)
+    db.add(post); db.commit(); db.refresh(post)
+    return post
+
+
+@app.get("/posts", response_model=List[PostOut])
+def list_posts(limit: int = 20, db: Session = Depends(get_db)):
+    return db.query(PostDB).order_by(PostDB.created_at.desc()).limit(limit).all()
+