Added checking for zoom link and saving the event

Author: Shakedp

app/routers/event.py

@@ -1,7 +1,15 @@
-from fastapi import APIRouter, Request
+import re
+from datetime import datetime as dt
+from fastapi import APIRouter, Request, Depends, HTTPException
+from fastapi.responses import RedirectResponse
+from starlette.status import HTTP_303_SEE_OTHER
 
+from app.database.database import get_db
+from app.database.models import Event, User
 from app.dependencies import templates
 
+ZOOM_REGEX = re.compile('https://.*?\.zoom.us/[a-z]/.[^.,\b\t\n]+')
+
 router = APIRouter(
     prefix="/event",
     tags=["event"],
@@ -15,6 +23,28 @@ async def eventedit(request: Request):
                                       {"request": request})
 
 
+@router.post("/edit")
+async def create_event(request: Request, session=Depends(get_db)):
+    data = await request.form()
+    print(data)
+    title = data['title']
+    content = data['description']
+    start = dt.strptime(data['start_date'] + ' ' + data['start_time'], '%Y-%m-%d %H:%M')
+    end = dt.strptime(data['end_date'] + ' ' + data['end_time'], '%Y-%m-%d %H:%M')
+    user = session.query(User).filter_by(id=1).first()
+    owner_id = user.id
+    location_type = data['location_type']
+    is_zoom = location_type == 'vc_url'
+    location = data['location']
+
+    if is_zoom and not ZOOM_REGEX.findall(location):
+        raise HTTPException(status_code=400, detail="VC type with no valid zoom link")
+    event = Event(title=title, content=content, start=start, end=end, owner_id=owner_id)
+    session.add(event)
+    session.commit()
+    return RedirectResponse(f'/event/view/{event.id}', status_code=HTTP_303_SEE_OTHER)
+
+
 @router.get("/view/{id}")
 async def eventview(request: Request, id: int):
     return templates.TemplateResponse("event/eventview.html",

tests/test_event.py

@@ -1,7 +1,36 @@
 from fastapi.testclient import TestClient
+from starlette.status import HTTP_303_SEE_OTHER
 
 from app.main import app
 
+CORRECT_EVENT_FORM_DATA = {
+    'title': 'test title',
+    'start_date': '2021-01-28',
+    'start_time': '15:59',
+    'end_date': '2021-01-27',
+    'end_time': '15:01',
+    'location_type': 'vc_url',
+    'location': 'https://us02web.zoom.us/j/875384596',
+    'description': 'content',
+    'color': 'red',
+    'availability': 'busy',
+    'privacy': 'public'
+}
+
+WRONG_EVENT_FORM_DATA = {
+    'title': 'test title',
+    'start_date': '2021-01-28',
+    'start_time': '15:59',
+    'end_date': '2021-01-27',
+    'end_time': '15:01',
+    'location_type': 'vc_url',
+    'location': 'not a zoom link',
+    'description': 'content',
+    'color': 'red',
+    'availability': 'busy',
+    'privacy': 'public'
+}
+
 client = TestClient(app)
 
 
@@ -11,6 +40,17 @@ def test_eventedit():
     assert b"Edit Event" in response.content
 
 
+def test_eventedit_post_correct(user):
+    response = client.post("/event/edit", data=CORRECT_EVENT_FORM_DATA)
+    assert response.status_code == HTTP_303_SEE_OTHER
+    assert '/event/view/' in response.headers['location']
+
+
+def test_eventedit_post_wrong(user):
+    response = client.post("/event/edit", data=WRONG_EVENT_FORM_DATA)
+    assert response.json()['detail'] == 'VC type with no valid zoom link'
+
+
 def test_eventview_with_id():
     response = client.get("/event/view/1")
     assert response.status_code == 200