June 17th update group chat

kevinamstutz · kevinamstutz · commit 4404899ee68d · 2022-06-17T15:29:23.000-04:00
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+.env
+__pycache__
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,8 @@
 FROM python:3.9-slim-bullseye
 
-RUN python -m pip install fastapi "uvicorn[standard]" jinja2 aiosql
+RUN apt update -y && apt install build-essential libpq-dev -y 
+
+RUN python -m pip install fastapi "uvicorn[standard]" jinja2 aiosql psycopg2-binary --no-binary psycopg2-binary python-dotenv
 
 WORKDIR /workspace
 
diff --git a/queries.sql b/queries.sql
@@ -0,0 +1,3 @@
+-- name: get-all-companies
+-- Get a list of all of the companies.
+SELECT company_name FROM companies;
diff --git a/templates/companies.html b/templates/companies.html
@@ -0,0 +1,12 @@
+<html>
+  <head>
+    <title>HOME</title>
+  </head>
+  <body>
+    <ul>
+      {% for company in payload1 %}
+      <li>{{ company }}</li>
+      {% endfor %}
+    </ul>
+  </body>
+</html>
diff --git a/website/main.py b/website/main.py
@@ -1,6 +1,11 @@
 from fastapi import FastAPI
 from fastapi.templating import Jinja2Templates
 from fastapi.requests import Request
+from dotenv import load_dotenv
+import psycopg2
+import aiosql
+import os
+
 
 app = FastAPI()
 templates = Jinja2Templates(directory='templates/')
@@ -17,3 +22,23 @@ async def root(request: Request):
         response = templates.TemplateResponse("index.html", {"request": request, "payload1": hello_world}) 
         return response
 
+@app.get("/companies")
+async def get_companies(request: Request):
+    load_dotenv()
+
+    conn = psycopg2.connect(f"dbname=tdm_main user=rdonly password={os.getenv('PG_PASSWORD')} host=db.tdm.geddes.rcac.purdue.edu")
+    queries = aiosql.from_path("queries.sql", "psycopg2")
+
+    companies = [v[0] for v in queries.get_all_companies(conn)]
+    print(type(companies))
+    print(companies)
+
+    accept = request.headers.get("accept")
+    hello_world = {"message": "hello world"}
+
+    if accept.split("/")[1] == 'json':
+        return hello_world
+
+    if len(accept.split(",")) > 1 or accept.split("/")[1] == 'html':
+        response = templates.TemplateResponse("companies.html", {"request": request, "payload1": companies}) 
+        return response

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+-- name: get-all-companies`
	`2`	`+-- Get a list of all of the companies.`
	`3`	`+SELECT company_name FROM companies;`