225 lines
9.3 KiB
Python
225 lines
9.3 KiB
Python
import re
|
|
import pickle
|
|
from secrets import token_urlsafe
|
|
from shutil import move
|
|
from typing import Union
|
|
from magic import Magic
|
|
from datetime import datetime, timezone
|
|
from os import makedirs, path, remove
|
|
from classes.exceptions import AlbumNameNotFoundError, SearchPageInvalidError, SearchTokenInvalidError, VideoNotFoundError, VideoSearchQueryEmptyError
|
|
from classes.models import Video, SearchResultsVideo, VideoPublic
|
|
from modules.security import User, get_current_active_user
|
|
from modules.app import app
|
|
from modules.database import col_videos, col_albums, col_tokens
|
|
from bson.objectid import ObjectId
|
|
from bson.errors import InvalidId
|
|
from pymongo import DESCENDING
|
|
|
|
from fastapi import UploadFile, Security
|
|
from fastapi.responses import UJSONResponse, Response
|
|
from starlette.status import HTTP_204_NO_CONTENT
|
|
|
|
video_post_responses = {
|
|
404: AlbumNameNotFoundError("name").openapi
|
|
}
|
|
@app.post("/albums/{album}/videos", description="Upload a video to album", response_class=UJSONResponse, response_model=Video, responses=video_post_responses)
|
|
async def video_upload(file: UploadFile, album: str, caption: Union[str, None] = None, current_user: User = Security(get_current_active_user, scopes=["videos.write"])):
|
|
|
|
if col_albums.find_one( {"user": current_user.user, "name": album} ) is None:
|
|
raise AlbumNameNotFoundError(album)
|
|
|
|
makedirs(path.join("data", "users", current_user.user, "albums", album), exist_ok=True)
|
|
|
|
filename = file.filename
|
|
|
|
if path.exists(path.join("data", "users", current_user.user, "albums", album, file.filename)):
|
|
base_name = file.filename.split(".")[:-1]
|
|
extension = file.filename.split(".")[-1]
|
|
filename = ".".join(base_name)+f"_{int(datetime.now().timestamp())}."+extension
|
|
|
|
with open(path.join("data", "users", current_user.user, "albums", album, filename), "wb") as f:
|
|
f.write(await file.read())
|
|
|
|
# Hashing and duplicates check should be here
|
|
|
|
# Coords extraction should be here
|
|
|
|
uploaded = col_videos.insert_one(
|
|
{
|
|
"user": current_user.user,
|
|
"album": album,
|
|
"filename": filename,
|
|
"dates": {
|
|
"uploaded": datetime.now(tz=timezone.utc),
|
|
"modified": datetime.now(tz=timezone.utc)
|
|
},
|
|
"caption": caption
|
|
}
|
|
)
|
|
|
|
return UJSONResponse(
|
|
{
|
|
"id": uploaded.inserted_id.__str__(),
|
|
"album": album,
|
|
"hash": "", # SHOULD BE DONE
|
|
"filename": filename
|
|
}
|
|
)
|
|
|
|
video_get_responses = {
|
|
404: VideoNotFoundError("id").openapi
|
|
}
|
|
@app.get("/videos/{id}", description="Get a video by id", responses=video_get_responses)
|
|
async def video_get(id: str, current_user: User = Security(get_current_active_user, scopes=["videos.read"])):
|
|
|
|
try:
|
|
video = col_videos.find_one( {"_id": ObjectId(id)} )
|
|
if video is None:
|
|
raise InvalidId(id)
|
|
except InvalidId:
|
|
raise VideoNotFoundError(id)
|
|
|
|
video_path = path.join("data", "users", current_user.user, "albums", video["album"], video["filename"])
|
|
|
|
mime = Magic(mime=True).from_file(video_path)
|
|
|
|
with open(video_path, "rb") as f: video_file = f.read()
|
|
|
|
return Response(video_file, media_type=mime)
|
|
|
|
video_move_responses = {
|
|
404: VideoNotFoundError("id").openapi
|
|
}
|
|
@app.put("/videos/{id}", description="Move a video into another album", response_model=VideoPublic, responses=video_move_responses)
|
|
async def video_move(id: str, album: str, current_user: User = Security(get_current_active_user, scopes=["videos.write"])):
|
|
|
|
try:
|
|
video = col_videos.find_one( {"_id": ObjectId(id)} )
|
|
if video is None:
|
|
raise InvalidId(id)
|
|
except InvalidId:
|
|
raise VideoNotFoundError(id)
|
|
|
|
if col_albums.find_one( {"user": current_user.user, "name": album} ) is None:
|
|
raise AlbumNameNotFoundError(album)
|
|
|
|
if path.exists(path.join("data", "users", current_user.user, "albums", album, video["filename"])):
|
|
base_name = video["filename"].split(".")[:-1]
|
|
extension = video["filename"].split(".")[-1]
|
|
filename = ".".join(base_name)+f"_{int(datetime.now().timestamp())}."+extension
|
|
else:
|
|
filename = video["filename"]
|
|
|
|
col_videos.find_one_and_update( {"_id": ObjectId(id)}, {"$set": {"album": album, "filename": filename, "dates.modified": datetime.now(tz=timezone.utc)}} )
|
|
|
|
move(
|
|
path.join("data", "users", current_user.user, "albums", video["album"], video["filename"]),
|
|
path.join("data", "users", current_user.user, "albums", album, filename)
|
|
)
|
|
|
|
return UJSONResponse(
|
|
{
|
|
"id": video["_id"].__str__(),
|
|
"caption": video["caption"],
|
|
"filename": filename
|
|
}
|
|
)
|
|
|
|
video_patch_responses = {
|
|
404: VideoNotFoundError("id").openapi
|
|
}
|
|
@app.patch("/videos/{id}", description="Change properties of a video", response_model=VideoPublic, responses=video_patch_responses)
|
|
async def video_patch(id: str, caption: str, current_user: User = Security(get_current_active_user, scopes=["videos.write"])):
|
|
|
|
try:
|
|
video = col_videos.find_one( {"_id": ObjectId(id)} )
|
|
if video is None:
|
|
raise InvalidId(id)
|
|
except InvalidId:
|
|
raise VideoNotFoundError(id)
|
|
|
|
col_videos.find_one_and_update( {"_id": ObjectId(id)}, {"$set": {"caption": caption, "dates.modified": datetime.now(tz=timezone.utc)}} )
|
|
|
|
return UJSONResponse(
|
|
{
|
|
"id": video["_id"].__str__(),
|
|
"caption": video["caption"],
|
|
"filename": video["filename"]
|
|
}
|
|
)
|
|
|
|
video_delete_responses = {
|
|
404: VideoNotFoundError("id").openapi
|
|
}
|
|
@app.delete("/videos/{id}", description="Delete a video by id", status_code=HTTP_204_NO_CONTENT, responses=video_delete_responses)
|
|
async def video_delete(id: str, current_user: User = Security(get_current_active_user, scopes=["videos.write"])):
|
|
|
|
try:
|
|
video = col_videos.find_one_and_delete( {"_id": ObjectId(id)} )
|
|
if video is None:
|
|
raise InvalidId(id)
|
|
except InvalidId:
|
|
raise VideoNotFoundError(id)
|
|
|
|
album = col_albums.find_one( {"name": video["album"]} )
|
|
|
|
remove(path.join("data", "users", current_user.user, "albums", video["album"], video["filename"]))
|
|
|
|
return Response(status_code=HTTP_204_NO_CONTENT)
|
|
|
|
video_find_responses = {
|
|
400: SearchPageInvalidError().openapi,
|
|
404: AlbumNameNotFoundError("name").openapi,
|
|
422: VideoSearchQueryEmptyError().openapi
|
|
}
|
|
@app.get("/albums/{album}/videos", description="Find a video by filename", response_class=UJSONResponse, response_model=SearchResultsVideo, responses=video_find_responses)
|
|
async def video_find(album: str, q: Union[str, None] = None, caption: Union[str, None] = None, page: int = 1, page_size: int = 100, current_user: User = Security(get_current_active_user, scopes=["videos.list"])):
|
|
|
|
if col_albums.find_one( {"user": current_user.user, "name": album} ) is None:
|
|
raise AlbumNameNotFoundError(album)
|
|
|
|
if page <= 0 or page_size <= 0:
|
|
raise SearchPageInvalidError()
|
|
|
|
output = {"results": []}
|
|
skip = (page-1)*page_size
|
|
|
|
if q is None and caption is None:
|
|
raise VideoSearchQueryEmptyError()
|
|
|
|
if q is None and caption is not None:
|
|
db_query = {"user": current_user.user, "album": album, "caption": re.compile(caption)}
|
|
db_query_count = {"user": current_user.user, "album": album, "caption": re.compile(caption)}
|
|
elif q is not None and caption is None:
|
|
db_query = list(col_videos.find({"user": current_user.user, "album": album, "filename": re.compile(q)}, limit=page_size, skip=skip).sort('dates.uploaded', DESCENDING))
|
|
db_query_count = {"user": current_user.user, "album": album, "caption": re.compile(q)}
|
|
else:
|
|
db_query = list(col_videos.find({"user": current_user.user, "album": album, "filename": re.compile(q), "caption": re.compile(caption)}, limit=page_size, skip=skip).sort('dates.uploaded', DESCENDING)) # type: ignore
|
|
db_query_count = {"user": current_user.user, "album": album, "filename": re.compile(q), "caption": re.compile(caption)} # type: ignore
|
|
|
|
videos = list(col_videos.find(db_query, limit=page_size, skip=skip).sort('dates.uploaded', DESCENDING))
|
|
|
|
for video in videos:
|
|
output["results"].append({"id": video["_id"].__str__(), "filename": video["filename"], "caption": video["caption"]})
|
|
|
|
if col_videos.count_documents( db_query_count ) > page*page_size:
|
|
token = str(token_urlsafe(32))
|
|
col_tokens.insert_one( {"token": token, "query": q, "album": album, "page": page+1, "page_size": page_size, "user": pickle.dumps(current_user)} )
|
|
output["next_page"] = f"/albums/{album}/videos/token?token={token}" # type: ignore
|
|
else:
|
|
output["next_page"] = None # type: ignore
|
|
|
|
return UJSONResponse(output)
|
|
|
|
video_find_token_responses = {
|
|
401: SearchTokenInvalidError().openapi
|
|
}
|
|
@app.get("/albums/{album}/videos/token", description="Find a video by token", response_class=UJSONResponse, response_model=SearchResultsVideo, responses=video_find_token_responses)
|
|
async def video_find_token(token: str):
|
|
|
|
found_record = col_tokens.find_one( {"token": token} )
|
|
|
|
if found_record is None:
|
|
raise SearchTokenInvalidError()
|
|
|
|
return await video_find(q=found_record["query"], album=found_record["album"], page=found_record["page"], page_size=found_record["page_size"], current_user=pickle.loads(found_record["user"])) |