added live urls for in and out

.gitignore

@@ -2,7 +2,7 @@ __pycache__
 log.txt
 env3.*
 env
+.env
 docs_build
 site_build
-venv
-test.db
+venv

core/utils.py

@@ -3,7 +3,7 @@ from sqlmodel import Session, SQLModel, create_engine
 from passlib.context import CryptContext
 from jose import jwt
 from config import settings
-
+import requests
 
 db_url = settings.DB_SQLITE
 
@@ -44,5 +44,36 @@ def create_access_token(data: dict, expires_delta: timedelta | None = None):
     encoded_jwt = jwt.encode(to_encode, settings.SECRET_KEY, algorithm=settings.ALGORITHM)
     return encoded_jwt
 
+
+
 def diff_time_stamp(mins: int):
-    return datetime.now()+timedelta(minutes=mins)
+    return datetime.now()+timedelta(minutes=mins)
+
+
+
+def get_key():
+    response = requests.post('https://open.ezvizlife.com/api/lapp/token/get', 
+                             headers={"Content-Type": "application/x-www-form-urlencoded"},
+                             data={"appKey": settings.EZV_KEY,
+                                   "appSecret": settings.EZV_SECRET}
+                             )
+    if response.status_code == 200:
+        return response.json()["data"]["accessToken"]
+    else:
+        return ""
+    
+def get_feed_url(accessToken, deviceSerial, code=654321, expireTime=90000, channelNo=1, protocol=2):
+    response = requests.post('https://isgpopen.ezvizlife.com/api/lapp/live/address/get', 
+                             headers={"Content-Type": "application/x-www-form-urlencoded"},
+                             data={"accessToken": accessToken,
+                                   "deviceSerial": deviceSerial,
+                                   "code": code,
+                                   "expireTime": expireTime,
+                                   "channelNo": channelNo,
+                                   "protocol": protocol,
+                                   }
+                             )
+    if response.status_code == 200:
+        return response.json()["data"]["url"]
+    else:
+        return ""

main.py

@@ -75,7 +75,7 @@ def seed_db(*,
     u3 = User(username="cs_is_admin", password=utils.get_password_hash("csisa"), disabled=False)
     u4 = User(username="ee_admin", password=utils.get_password_hash("eea"), disabled=False)
     s1 = Site(name="CS Dept", location="MCS", in_camera="2a")
-    s2 = Site(name="IS Dept", location="NRB", in_camera="3a")
+    s2 = Site(name="IS Dept", location="NRB", in_camera="AA4823505", out_camera="AB9438217")
     s3 = Site(name="EE Dept", location="MCS", in_camera="1a")
     u1.sites.append(s1)
     u2.sites.append(s2)
@@ -132,4 +132,12 @@ def seed_db(*,
     session.add(Visit(time_in=utils.diff_time_stamp(65).time(), time_out=utils.diff_time_stamp(10).time(), is_group=True, is_female=False, is_new=True, site_id=2))
     session.add(Visit(time_in=utils.diff_time_stamp(68).time(), time_out=utils.diff_time_stamp(9).time(), is_group=False, is_female=False, is_new=True, site_id=2))
     session.add(Visit(time_in=utils.diff_time_stamp(70).time(), time_out=utils.diff_time_stamp(7).time(), is_group=False, is_female=True, is_new=True, site_id=2))
-    session.commit()
+    session.commit()
+
+
+
+# @app.get("/feed_url/")
+# def get_feed(*,
+#     session: Session = Depends(utils.get_session), 
+#     user: Annotated[User, Depends(crud.get_current_super_user)],
+#     ):

models.py

@@ -85,6 +85,8 @@ class SiteBase(SQLModel):
     contact: str | None = None
     in_camera: str | None = Field(default=None, unique=True)
     out_camera: str | None = Field(default=None, unique=True)
+    in_url: str | None
+    out_url: str | None
 
 class Site(SiteBase, table=True):
     id: int | None = Field(default=None, primary_key=True)

routers/profile.py

@@ -75,7 +75,15 @@ def read_site(*,
               current_user: Annotated[User, Depends(crud.get_current_active_user)], 
               site_id: int
               ):
-    return crud.get_current_site(session, current_user, site_id)
+    current_site = crud.get_current_site(session, current_user, site_id)
+    access_token = utils.get_key()
+    in_url= utils.get_feed_url(access_token, current_site.in_camera)
+    out_url= utils.get_feed_url(access_token, current_site.out_camera)
+    current_site.in_url = in_url
+    current_site.out_url = out_url
+    session.add(current_site)
+    session.commit()
+    return current_site