move contact api endpoint to contact route

Author: surajit

main.py

@@ -6,17 +6,16 @@
 from sqlalchemy.orm import Session
 
 from crud.company import CompanyExistException
-from crud.contact import ContactExistException
 from dependencies.dependencies import get_db
 from schemas import schema
 from hubspot_api import utils
 from crud import company as _company
-from crud import contact as _contact
 from routes import email
+from routes import contact
 
 app = FastAPI()
 app.include_router(email.router)
-
+app.include_router(contact.router)
 templates = Jinja2Templates(directory="templates")
 
 
@@ -27,24 +26,6 @@ def read_root(request: Request):
     )
 
 
-@app.post("/contacts")
-def create_contact(contact: schema.CreateContact, db: Session = Depends(get_db)):
-    db_contact = None
-    hubspot_contact = None
-    try:
-        org_name = contact.company_name
-        db_org = _company.filter_company_by_name(db, org_name)
-        if db_org:
-            org = db_org.all()[0]
-            db_contact = _contact.create_contact(db, contact, org.id)
-            hubspot_contact = utils.create_contact(data=contact.dict())
-    except (utils.ContactException, ContactExistException) as exc:
-        if isinstance(exc, utils.ContactException) and db_contact:
-            _contact.delete_contact(db, db_contact.id)
-        raise HTTPException(status_code=200, detail=str(exc))
-    return hubspot_contact
-
-
 @app.post("/companies")
 def create_company(company: schema.CreateCompany, db: Session = Depends(get_db)):
     db_company = None
@@ -65,12 +46,3 @@ def get_company(company_name):
     except utils.CompanyException as exc:
         raise HTTPException(status_code=200, detail=str(exc))
     return company
-
-
-@app.get("/contact/{email}/")
-def get_contact(email):
-    try:
-        contact = utils.get_contact_by_email(email)
-    except utils.ContactException as exc:
-        raise HTTPException(status_code=200, detail=str(exc))
-    return contact

routes/contact.py

@@ -1 +1,43 @@
-# TODO - Move contact routes from main.py to here
+from fastapi import APIRouter
+from fastapi import Depends
+from fastapi import HTTPException
+from sqlalchemy.orm import Session
+
+from crud.contact import ContactExistException
+from crud import contact as _contact
+from schemas import schema
+from hubspot_api import utils
+from dependencies.dependencies import get_db
+from crud import company as _company
+
+router = APIRouter(
+    tags=["contact"],
+    responses={404: {"description": "Not found"}},
+)
+
+
+@router.post("/contacts/")
+def create_contact(contact: schema.CreateContact, db: Session = Depends(get_db)):
+    db_contact = None
+    hubspot_contact = None
+    try:
+        org_name = contact.company_name
+        db_org = _company.filter_company_by_name(db, org_name)
+        if db_org:
+            org = db_org.all()[0]
+            db_contact = _contact.create_contact(db, contact, org.id)
+            hubspot_contact = utils.create_contact(data=contact.dict())
+    except (utils.ContactException, ContactExistException) as exc:
+        if isinstance(exc, utils.ContactException) and db_contact:
+            _contact.delete_contact(db, db_contact.id)
+        raise HTTPException(status_code=200, detail=str(exc))
+    return hubspot_contact
+
+
+@router.get("/contact/{email}/")
+def get_contact(email):
+    try:
+        contact = utils.get_contact_by_email(email)
+    except utils.ContactException as exc:
+        raise HTTPException(status_code=200, detail=str(exc))
+    return contact