From 0763b110cbe8fe7eedb99749742a395db751e605 Mon Sep 17 00:00:00 2001
From: Colin Bassett <tropii@Colins-MacBook.local>
Date: Fri, 21 Feb 2025 13:29:16 -0500
Subject: [PATCH] Implement a SQL Database

---
 database.db | Bin 0 -> 12288 bytes
 main.py     |  49 ++++++++++++++++++++++++++++++++++++++++++++++---
 2 files changed, 46 insertions(+), 3 deletions(-)
 create mode 100644 database.db

diff --git a/database.db b/database.db
new file mode 100644
index 0000000000000000000000000000000000000000..99b4e6718a4a7462c2a95acdc98aeee74fee8e76
GIT binary patch
literal 12288
zcmeI#K}*9h6u|Mc6J-UNm$~cnFjzsn_yw$6&|&SgR&k(=N~;vyCR^2A_3D>4kqsuy
zyNrJz%?t0nKz=>t>M|_KTpq`hVOq+e+EI?9jzpAFu9+1x7sKv178|oVOYvW!s}A12
zwoTTp?W)?H`7jVb009ILKmY**5I_I{1Q7T)0v&g=(X2c7#VE^X#cc38%_oCVI?QcQ
zX~#N9v~;76zLn*y^dhkp(z3Pw$DXs*sMnp(W|=-b|Mc3dJMT||<517*y=sw}0~39s
zV~Ki+MAu=sFJ9FiBsW3aJ`H{+(x-f7!%I4yzK<u_%BG9BdmhAnIn#Y<6`Ak(_HAw+
z{R{*UKmY**5I_I{1Q0*~0R#|OV*$_Jjrac=|6#6%00IagfB*srAb<b@2q1ufQr~kv
BO8Ec)

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
index d2957b3..c3de155 100644
--- a/main.py
+++ b/main.py
@@ -1,9 +1,43 @@
-from fastapi import FastAPI
+# Imports
+from typing import Annotated
+
+from fastapi import FastAPI, Depends, HTTPException, Query
+from sqlmodel import Field, Session, SQLModel, create_engine, select
 import hashlib
 import random
 
+# Initialize the FastAPI App
 app = FastAPI()
 
+# Create the user table
+
+class User(SQLModel, table=True):
+    id: int = Field(default=None, primary_key=True)
+    name: str = Field(index=True)
+    age: int
+    password: str
+
+# SQLModel stuff
+
+sqlite_file_name = "database.db"
+sqlite_url = f"sqlite:///{sqlite_file_name}"
+
+connect_args = {"check_same_thread": False}
+engine = create_engine(sqlite_url, connect_args=connect_args)
+
+def create_db_and_tables():
+    SQLModel.metadata.create_all(engine)
+
+def get_session():
+    with Session(engine) as session:
+        yield session
+
+SessionDep = Annotated[Session, Depends(get_session)]
+
+@app.on_event("startup")
+def on_startup():
+    create_db_and_tables()
+
 @app.get("/")
 def hello_world():
     return {"message": "Hello World!"}
@@ -17,9 +51,18 @@ def hash(password: str):
 
 @app.get("/verify/{password}/{hash}/{salt}")
 def verify(password: str, hash: str, salt: int):
-    password = password + str(salt)
-    hashed = hashlib.sha256(password.encode()).hexdigest()
+    if salt != 0:
+        password = password + str(salt)
+        hashed = hashlib.sha256(password.encode()).hexdigest()
+    else:
+        hashed = hashlib.sha256(password.encode()).hexdigest()
+
     if hashed == hash:
         return {"message": "Password is correct", "correct": True}
 
     return {"message": "Password is incorrect", "correct": False}
+
+@app.get("/hash/no-salt/{password}")
+def no_salt(password: str):
+    hashed = hashlib.sha256(password.encode()).hexdigest()
+    return {"hash": hashed}