Compare commits
38 Commits
Author | SHA1 | Date | |
---|---|---|---|
6be51c5aaa | |||
840e3022b3 | |||
24f4773dd7
|
|||
00d3d62762
|
|||
2a29b85ad2 | |||
9bdc788078
|
|||
5a5103ea9c | |||
ccf4c43bb9 | |||
19e0531a24
|
|||
b46f3fb0fd
|
|||
d2f3d7e687
|
|||
83dd4b6746
|
|||
47435c6128
|
|||
db77f62459
|
|||
b51026b200 | |||
b30547eca8 | |||
782b489db2 | |||
d085a0e639 | |||
30d72c84ed | |||
e1e42fdb60 | |||
36169b0e77 | |||
5de935cd21 | |||
1e6afc6b0c | |||
f9e6ee9c72 | |||
f512df408f | |||
aa083811dc | |||
4d24696d3d | |||
c7cb4a6dff | |||
4060aae038 | |||
4eea82a160 | |||
4ce4264580 | |||
6feed4359a | |||
2afc82cf01 | |||
bf0046c3d5 | |||
c55a2d0d44 | |||
a380da81bb | |||
e858e7d7f4 | |||
fcbbd4f2bf |
20
.renovaterc
Normal file
20
.renovaterc
Normal file
@@ -0,0 +1,20 @@
|
||||
{
|
||||
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
|
||||
"extends": [
|
||||
"config:base"
|
||||
],
|
||||
"baseBranches": [
|
||||
"dev"
|
||||
],
|
||||
"packageRules": [
|
||||
{
|
||||
"matchUpdateTypes": [
|
||||
"minor",
|
||||
"patch",
|
||||
"pin",
|
||||
"digest"
|
||||
],
|
||||
"automerge": true
|
||||
}
|
||||
]
|
||||
}
|
@@ -9,7 +9,7 @@ Small and simple API server for saving photos and videos.
|
||||
|
||||
## Dependencies
|
||||
|
||||
* [Python 3.7+](https://www.python.org) (3.9+ recommended)
|
||||
* [Python 3.8+](https://www.python.org) (3.9+ recommended)
|
||||
* [MongoDB](https://www.mongodb.com)
|
||||
* [exiftool](https://exiftool.org)
|
||||
* [jpegoptim](https://github.com/tjko/jpegoptim)
|
||||
|
@@ -1,7 +1,9 @@
|
||||
from typing import Literal
|
||||
|
||||
from fastapi import HTTPException
|
||||
|
||||
class AlbumNotFoundError(Exception):
|
||||
|
||||
class AlbumNotFoundError(HTTPException):
|
||||
"""Raises HTTP 404 if no album with this ID found."""
|
||||
|
||||
def __init__(self, id: str):
|
||||
@@ -16,7 +18,7 @@ class AlbumNotFoundError(Exception):
|
||||
}
|
||||
|
||||
|
||||
class AlbumNameNotFoundError(Exception):
|
||||
class AlbumNameNotFoundError(HTTPException):
|
||||
"""Raises HTTP 404 if no album with this name found."""
|
||||
|
||||
def __init__(self, name: str):
|
||||
@@ -29,9 +31,15 @@ class AlbumNameNotFoundError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=404,
|
||||
detail=self.openapi["content"]["application/json"]["example"][
|
||||
"detail"
|
||||
].format(name=self.name),
|
||||
)
|
||||
|
||||
|
||||
class AlbumAlreadyExistsError(Exception):
|
||||
class AlbumAlreadyExistsError(HTTPException):
|
||||
"""Raises HTTP 409 if album with this name already exists."""
|
||||
|
||||
def __init__(self, name: str):
|
||||
@@ -44,9 +52,15 @@ class AlbumAlreadyExistsError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=409,
|
||||
detail=self.openapi["content"]["application/json"]["example"][
|
||||
"detail"
|
||||
].format(name=self.name),
|
||||
)
|
||||
|
||||
|
||||
class AlbumIncorrectError(Exception):
|
||||
class AlbumIncorrectError(HTTPException):
|
||||
"""Raises HTTP 406 if album's title or name is invalid."""
|
||||
|
||||
def __init__(self, place: Literal["name", "title"], error: str) -> None:
|
||||
@@ -56,13 +70,19 @@ class AlbumIncorrectError(Exception):
|
||||
"description": "Album Name/Title Invalid",
|
||||
"content": {
|
||||
"application/json": {
|
||||
"example": {"detail": "Album {name/title} invalid: {error}"}
|
||||
"example": {"detail": "Album {place} invalid: {error}"}
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=406,
|
||||
detail=self.openapi["content"]["application/json"]["example"][
|
||||
"detail"
|
||||
].format(place=self.place, error=self.error),
|
||||
)
|
||||
|
||||
|
||||
class PhotoNotFoundError(Exception):
|
||||
class PhotoNotFoundError(HTTPException):
|
||||
"""Raises HTTP 404 if no photo with this ID found."""
|
||||
|
||||
def __init__(self, id: str):
|
||||
@@ -75,9 +95,15 @@ class PhotoNotFoundError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=404,
|
||||
detail=self.openapi["content"]["application/json"]["example"][
|
||||
"detail"
|
||||
].format(id=self.id),
|
||||
)
|
||||
|
||||
|
||||
class PhotoSearchQueryEmptyError(Exception):
|
||||
class PhotoSearchQueryEmptyError(HTTPException):
|
||||
"""Raises HTTP 422 if no photo search query provided."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -91,9 +117,13 @@ class PhotoSearchQueryEmptyError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=422,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class VideoNotFoundError(Exception):
|
||||
class VideoNotFoundError(HTTPException):
|
||||
"""Raises HTTP 404 if no video with this ID found."""
|
||||
|
||||
def __init__(self, id: str):
|
||||
@@ -106,9 +136,15 @@ class VideoNotFoundError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=404,
|
||||
detail=self.openapi["content"]["application/json"]["example"][
|
||||
"detail"
|
||||
].format(id=self.id),
|
||||
)
|
||||
|
||||
|
||||
class VideoSearchQueryEmptyError(Exception):
|
||||
class VideoSearchQueryEmptyError(HTTPException):
|
||||
"""Raises HTTP 422 if no video search query provided."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -122,9 +158,13 @@ class VideoSearchQueryEmptyError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=422,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class SearchPageInvalidError(Exception):
|
||||
class SearchPageInvalidError(HTTPException):
|
||||
"""Raises HTTP 400 if page or page size are not in valid range."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -138,9 +178,13 @@ class SearchPageInvalidError(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=400,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class SearchTokenInvalidError(Exception):
|
||||
class SearchTokenInvalidError(HTTPException):
|
||||
"""Raises HTTP 401 if search token is not valid."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -150,9 +194,13 @@ class SearchTokenInvalidError(Exception):
|
||||
"application/json": {"example": {"detail": "Invalid search token."}}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=401,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class UserEmailCodeInvalid(Exception):
|
||||
class UserEmailCodeInvalid(HTTPException):
|
||||
"""Raises HTTP 400 if email confirmation code is not valid."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -164,9 +212,13 @@ class UserEmailCodeInvalid(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=400,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class UserAlreadyExists(Exception):
|
||||
class UserAlreadyExists(HTTPException):
|
||||
"""Raises HTTP 409 if user with this name already exists."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -178,9 +230,13 @@ class UserAlreadyExists(Exception):
|
||||
}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=409,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class AccessTokenInvalidError(Exception):
|
||||
class AccessTokenInvalidError(HTTPException):
|
||||
"""Raises HTTP 401 if access token is not valid."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -190,9 +246,13 @@ class AccessTokenInvalidError(Exception):
|
||||
"application/json": {"example": {"detail": "Invalid access token."}}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=401,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
||||
|
||||
class UserCredentialsInvalid(Exception):
|
||||
class UserCredentialsInvalid(HTTPException):
|
||||
"""Raises HTTP 401 if user credentials are not valid."""
|
||||
|
||||
def __init__(self):
|
||||
@@ -202,3 +262,7 @@ class UserCredentialsInvalid(Exception):
|
||||
"application/json": {"example": {"detail": "Invalid credentials."}}
|
||||
},
|
||||
}
|
||||
super().__init__(
|
||||
status_code=401,
|
||||
detail=self.openapi["content"]["application/json"]["example"]["detail"],
|
||||
)
|
||||
|
@@ -1,4 +1,5 @@
|
||||
from typing import List, Union
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
|
@@ -1,7 +1,5 @@
|
||||
from fastapi import Request
|
||||
from fastapi.responses import UJSONResponse
|
||||
from modules.app import app
|
||||
from classes.exceptions import *
|
||||
from starlette.status import (
|
||||
HTTP_400_BAD_REQUEST,
|
||||
HTTP_401_UNAUTHORIZED,
|
||||
@@ -11,6 +9,23 @@ from starlette.status import (
|
||||
HTTP_422_UNPROCESSABLE_ENTITY,
|
||||
)
|
||||
|
||||
from classes.exceptions import (
|
||||
AlbumNotFoundError,
|
||||
AlbumAlreadyExistsError,
|
||||
AlbumIncorrectError,
|
||||
PhotoNotFoundError,
|
||||
PhotoSearchQueryEmptyError,
|
||||
VideoNotFoundError,
|
||||
VideoSearchQueryEmptyError,
|
||||
SearchPageInvalidError,
|
||||
SearchTokenInvalidError,
|
||||
AccessTokenInvalidError,
|
||||
UserEmailCodeInvalid,
|
||||
UserAlreadyExists,
|
||||
UserCredentialsInvalid,
|
||||
)
|
||||
from modules.app import app
|
||||
|
||||
|
||||
@app.exception_handler(AlbumNotFoundError)
|
||||
async def album_not_found_exception_handler(request: Request, exc: AlbumNotFoundError):
|
||||
|
@@ -1,31 +1,42 @@
|
||||
from os import path
|
||||
from modules.app import app
|
||||
|
||||
import aiofiles
|
||||
from fastapi.responses import HTMLResponse, Response
|
||||
|
||||
from modules.app import app
|
||||
|
||||
|
||||
@app.get("/pages/matter.css", include_in_schema=False)
|
||||
async def page_matter():
|
||||
with open(path.join("pages", "matter.css"), "r", encoding="utf-8") as f:
|
||||
output = f.read()
|
||||
async with aiofiles.open(
|
||||
path.join("pages", "matter.css"), "r", encoding="utf-8"
|
||||
) as f:
|
||||
output = await f.read()
|
||||
return Response(content=output)
|
||||
|
||||
|
||||
@app.get("/pages/{page}/{file}", include_in_schema=False)
|
||||
async def page_assets(page: str, file: str):
|
||||
with open(path.join("pages", page, file), "r", encoding="utf-8") as f:
|
||||
output = f.read()
|
||||
async with aiofiles.open(
|
||||
path.join("pages", page, file), "r", encoding="utf-8"
|
||||
) as f:
|
||||
output = await f.read()
|
||||
return Response(content=output)
|
||||
|
||||
|
||||
@app.get("/", include_in_schema=False)
|
||||
async def page_home():
|
||||
with open(path.join("pages", "home", "index.html"), "r", encoding="utf-8") as f:
|
||||
output = f.read()
|
||||
async with aiofiles.open(
|
||||
path.join("pages", "home", "index.html"), "r", encoding="utf-8"
|
||||
) as f:
|
||||
output = await f.read()
|
||||
return HTMLResponse(content=output)
|
||||
|
||||
|
||||
@app.get("/register", include_in_schema=False)
|
||||
async def page_register():
|
||||
with open(path.join("pages", "register", "index.html"), "r", encoding="utf-8") as f:
|
||||
output = f.read()
|
||||
async with aiofiles.open(
|
||||
path.join("pages", "register", "index.html"), "r", encoding="utf-8"
|
||||
) as f:
|
||||
output = await f.read()
|
||||
return HTMLResponse(content=output)
|
||||
|
@@ -1,15 +1,24 @@
|
||||
import re
|
||||
import pickle
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from os import makedirs, path, remove, system
|
||||
from secrets import token_urlsafe
|
||||
from shutil import move
|
||||
from threading import Thread
|
||||
from typing import Union
|
||||
from uuid import uuid4
|
||||
from magic import Magic
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from os import makedirs, path, remove, system
|
||||
|
||||
import aiofiles
|
||||
from bson.errors import InvalidId
|
||||
from bson.objectid import ObjectId
|
||||
from fastapi import Security, UploadFile
|
||||
from fastapi.responses import Response, UJSONResponse
|
||||
from jose import JWTError, jwt
|
||||
from magic import Magic
|
||||
from plum.exceptions import UnpackError
|
||||
from pydantic import ValidationError
|
||||
from pymongo import DESCENDING
|
||||
from starlette.status import HTTP_204_NO_CONTENT, HTTP_409_CONFLICT
|
||||
|
||||
from classes.exceptions import (
|
||||
AccessTokenInvalidError,
|
||||
AlbumNameNotFoundError,
|
||||
@@ -19,8 +28,10 @@ from classes.exceptions import (
|
||||
SearchTokenInvalidError,
|
||||
)
|
||||
from classes.models import Photo, PhotoPublic, SearchResultsPhoto
|
||||
from modules.app import app
|
||||
from modules.database import col_albums, col_photos, col_tokens
|
||||
from modules.exif_reader import extract_location
|
||||
from modules.hasher import get_phash, get_duplicates
|
||||
from modules.hasher import get_duplicates, get_phash
|
||||
from modules.scheduler import scheduler
|
||||
from modules.security import (
|
||||
ALGORITHM,
|
||||
@@ -31,23 +42,6 @@ from modules.security import (
|
||||
get_current_active_user,
|
||||
get_user,
|
||||
)
|
||||
from modules.app import app
|
||||
from modules.database import col_photos, col_albums, col_tokens
|
||||
from pymongo import DESCENDING
|
||||
from bson.objectid import ObjectId
|
||||
from bson.errors import InvalidId
|
||||
from plum.exceptions import UnpackError
|
||||
from jose import JWTError, jwt
|
||||
|
||||
from fastapi import UploadFile, Security
|
||||
from fastapi.responses import UJSONResponse, Response
|
||||
from fastapi.exceptions import HTTPException
|
||||
from starlette.status import (
|
||||
HTTP_204_NO_CONTENT,
|
||||
HTTP_401_UNAUTHORIZED,
|
||||
HTTP_409_CONFLICT,
|
||||
)
|
||||
|
||||
from modules.utils import configGet, logWrite
|
||||
|
||||
|
||||
@@ -130,7 +124,7 @@ async def photo_upload(
|
||||
".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
|
||||
)
|
||||
|
||||
with open(
|
||||
async with aiofiles.open(
|
||||
path.join("data", "users", current_user.user, "albums", album, filename), "wb"
|
||||
) as f:
|
||||
f.write(await file.read())
|
||||
@@ -266,16 +260,34 @@ if configGet("media_token_access") is True:
|
||||
|
||||
mime = Magic(mime=True).from_file(image_path)
|
||||
|
||||
with open(image_path, "rb") as f:
|
||||
image_file = f.read()
|
||||
async with aiofiles.open(image_path, "rb") as f:
|
||||
image_file = await f.read()
|
||||
|
||||
return Response(image_file, media_type=mime)
|
||||
|
||||
|
||||
photo_get_responses = {404: PhotoNotFoundError("id").openapi}
|
||||
photo_get_responses = {
|
||||
200: {
|
||||
"content": {
|
||||
"application/octet-stream": {
|
||||
"schema": {
|
||||
"type": "string",
|
||||
"format": "binary",
|
||||
"contentMediaType": "image/*",
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
404: PhotoNotFoundError("id").openapi,
|
||||
}
|
||||
|
||||
|
||||
@app.get("/photos/{id}", description="Get a photo by id", responses=photo_get_responses)
|
||||
@app.get(
|
||||
"/photos/{id}",
|
||||
description="Get a photo by id",
|
||||
responses=photo_get_responses,
|
||||
response_class=Response,
|
||||
)
|
||||
async def photo_get(
|
||||
id: str,
|
||||
current_user: User = Security(get_current_active_user, scopes=["photos.read"]),
|
||||
@@ -293,8 +305,8 @@ async def photo_get(
|
||||
|
||||
mime = Magic(mime=True).from_file(image_path)
|
||||
|
||||
with open(image_path, "rb") as f:
|
||||
image_file = f.read()
|
||||
async with aiofiles.open(image_path, "rb") as f:
|
||||
image_file = await f.read()
|
||||
|
||||
return Response(image_file, media_type=mime)
|
||||
|
||||
@@ -444,6 +456,7 @@ async def photo_delete(
|
||||
|
||||
photo_find_responses = {
|
||||
400: SearchPageInvalidError().openapi,
|
||||
401: SearchTokenInvalidError().openapi,
|
||||
404: AlbumNameNotFoundError("name").openapi,
|
||||
422: PhotoSearchQueryEmptyError().openapi,
|
||||
}
|
||||
@@ -451,7 +464,7 @@ photo_find_responses = {
|
||||
|
||||
@app.get(
|
||||
"/albums/{album}/photos",
|
||||
description="Find a photo by filename",
|
||||
description="Find a photo by filename, caption, location or token",
|
||||
response_class=UJSONResponse,
|
||||
response_model=SearchResultsPhoto,
|
||||
responses=photo_find_responses,
|
||||
@@ -460,6 +473,7 @@ async def photo_find(
|
||||
album: str,
|
||||
q: Union[str, None] = None,
|
||||
caption: Union[str, None] = None,
|
||||
token: Union[str, None] = None,
|
||||
page: int = 1,
|
||||
page_size: int = 100,
|
||||
lat: Union[float, None] = None,
|
||||
@@ -467,6 +481,24 @@ async def photo_find(
|
||||
radius: Union[int, None] = None,
|
||||
current_user: User = Security(get_current_active_user, scopes=["photos.list"]),
|
||||
):
|
||||
if token is not None:
|
||||
found_record = col_tokens.find_one({"token": token})
|
||||
|
||||
if found_record is None:
|
||||
raise SearchTokenInvalidError()
|
||||
|
||||
return await photo_find(
|
||||
album=album,
|
||||
q=found_record["query"],
|
||||
caption=found_record["caption"],
|
||||
lat=found_record["lat"],
|
||||
lng=found_record["lng"],
|
||||
radius=found_record["radius"],
|
||||
page=found_record["page"],
|
||||
page_size=found_record["page_size"],
|
||||
current_user=current_user,
|
||||
)
|
||||
|
||||
if col_albums.find_one({"user": current_user.user, "name": album}) is None:
|
||||
raise AlbumNameNotFoundError(album)
|
||||
|
||||
@@ -543,39 +575,16 @@ async def photo_find(
|
||||
{
|
||||
"token": token,
|
||||
"query": q,
|
||||
"album": album,
|
||||
"caption": caption,
|
||||
"lat": lat,
|
||||
"lng": lng,
|
||||
"radius": radius,
|
||||
"page": page + 1,
|
||||
"page_size": page_size,
|
||||
"user": pickle.dumps(current_user),
|
||||
}
|
||||
)
|
||||
output["next_page"] = f"/albums/{album}/photos/token?token={token}" # type: ignore
|
||||
output["next_page"] = f"/albums/{album}/photos/?token={token}" # type: ignore
|
||||
else:
|
||||
output["next_page"] = None # type: ignore
|
||||
|
||||
return UJSONResponse(output)
|
||||
|
||||
|
||||
photo_find_token_responses = {401: SearchTokenInvalidError().openapi}
|
||||
|
||||
|
||||
@app.get(
|
||||
"/albums/{album}/photos/token",
|
||||
description="Find a photo by token",
|
||||
response_class=UJSONResponse,
|
||||
response_model=SearchResultsPhoto,
|
||||
responses=photo_find_token_responses,
|
||||
)
|
||||
async def photo_find_token(token: str):
|
||||
found_record = col_tokens.find_one({"token": token})
|
||||
|
||||
if found_record is None:
|
||||
raise SearchTokenInvalidError()
|
||||
|
||||
return await photo_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"]),
|
||||
)
|
||||
|
@@ -1,12 +1,10 @@
|
||||
from datetime import timedelta
|
||||
from classes.exceptions import UserCredentialsInvalid
|
||||
from modules.app import app
|
||||
|
||||
from fastapi import Depends
|
||||
from fastapi.security import (
|
||||
OAuth2PasswordRequestForm,
|
||||
)
|
||||
from fastapi.security import OAuth2PasswordRequestForm
|
||||
|
||||
from classes.exceptions import UserCredentialsInvalid
|
||||
from modules.app import app
|
||||
from modules.security import (
|
||||
ACCESS_TOKEN_EXPIRE_DAYS,
|
||||
Token,
|
||||
|
@@ -1,27 +1,19 @@
|
||||
from datetime import datetime, timedelta
|
||||
from uuid import uuid1
|
||||
|
||||
from fastapi import Depends, Form
|
||||
from fastapi.responses import Response, UJSONResponse
|
||||
from starlette.status import HTTP_204_NO_CONTENT
|
||||
|
||||
from classes.exceptions import (
|
||||
UserAlreadyExists,
|
||||
UserCredentialsInvalid,
|
||||
UserEmailCodeInvalid,
|
||||
)
|
||||
from modules.database import (
|
||||
col_users,
|
||||
col_albums,
|
||||
col_photos,
|
||||
col_emails,
|
||||
col_videos,
|
||||
col_emails,
|
||||
)
|
||||
from modules.app import app
|
||||
from modules.utils import configGet, logWrite
|
||||
from modules.scheduler import scheduler
|
||||
from modules.database import col_albums, col_emails, col_photos, col_users, col_videos
|
||||
from modules.mailer import mail_sender
|
||||
|
||||
from uuid import uuid1
|
||||
from fastapi import Depends, Form
|
||||
from fastapi.responses import Response, UJSONResponse
|
||||
from starlette.status import HTTP_204_NO_CONTENT
|
||||
|
||||
from modules.scheduler import scheduler
|
||||
from modules.security import (
|
||||
User,
|
||||
get_current_active_user,
|
||||
@@ -29,6 +21,7 @@ from modules.security import (
|
||||
get_user,
|
||||
verify_password,
|
||||
)
|
||||
from modules.utils import configGet, logWrite
|
||||
|
||||
|
||||
async def send_confirmation(user: str, email: str):
|
||||
|
@@ -1,11 +1,19 @@
|
||||
import re
|
||||
import pickle
|
||||
from datetime import datetime, timezone
|
||||
from os import makedirs, path, remove
|
||||
from secrets import token_urlsafe
|
||||
from shutil import move
|
||||
from typing import Union
|
||||
|
||||
import aiofiles
|
||||
from bson.errors import InvalidId
|
||||
from bson.objectid import ObjectId
|
||||
from fastapi import Security, UploadFile
|
||||
from fastapi.responses import Response, UJSONResponse
|
||||
from magic import Magic
|
||||
from datetime import datetime, timezone
|
||||
from os import makedirs, path, remove
|
||||
from pymongo import DESCENDING
|
||||
from starlette.status import HTTP_204_NO_CONTENT
|
||||
|
||||
from classes.exceptions import (
|
||||
AlbumNameNotFoundError,
|
||||
SearchPageInvalidError,
|
||||
@@ -13,17 +21,10 @@ from classes.exceptions import (
|
||||
VideoNotFoundError,
|
||||
VideoSearchQueryEmptyError,
|
||||
)
|
||||
from classes.models import Video, SearchResultsVideo, VideoPublic
|
||||
from modules.security import User, get_current_active_user
|
||||
from classes.models import SearchResultsVideo, Video, VideoPublic
|
||||
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
|
||||
from modules.database import col_albums, col_tokens, col_videos
|
||||
from modules.security import User, get_current_active_user
|
||||
|
||||
video_post_responses = {404: AlbumNameNotFoundError("name").openapi}
|
||||
|
||||
@@ -59,10 +60,10 @@ async def video_upload(
|
||||
".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
|
||||
)
|
||||
|
||||
with open(
|
||||
async with aiofiles.open(
|
||||
path.join("data", "users", current_user.user, "albums", album, filename), "wb"
|
||||
) as f:
|
||||
f.write(await file.read())
|
||||
await f.write(await file.read())
|
||||
|
||||
# Hashing and duplicates check should be here
|
||||
|
||||
@@ -91,10 +92,28 @@ async def video_upload(
|
||||
)
|
||||
|
||||
|
||||
video_get_responses = {404: VideoNotFoundError("id").openapi}
|
||||
video_get_responses = {
|
||||
200: {
|
||||
"content": {
|
||||
"application/octet-stream": {
|
||||
"schema": {
|
||||
"type": "string",
|
||||
"format": "binary",
|
||||
"contentMediaType": "video/*",
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
404: VideoNotFoundError("id").openapi,
|
||||
}
|
||||
|
||||
|
||||
@app.get("/videos/{id}", description="Get a video by id", responses=video_get_responses)
|
||||
@app.get(
|
||||
"/videos/{id}",
|
||||
description="Get a video by id",
|
||||
responses=video_get_responses,
|
||||
response_class=Response,
|
||||
)
|
||||
async def video_get(
|
||||
id: str,
|
||||
current_user: User = Security(get_current_active_user, scopes=["videos.read"]),
|
||||
@@ -112,10 +131,10 @@ async def video_get(
|
||||
|
||||
mime = Magic(mime=True).from_file(video_path)
|
||||
|
||||
with open(video_path, "rb") as f:
|
||||
video_file = f.read()
|
||||
async with aiofiles.open(video_path, "rb") as f:
|
||||
video_file = await f.read()
|
||||
|
||||
return Response(video_file, media_type=mime)
|
||||
return Response(content=video_file, media_type=mime)
|
||||
|
||||
|
||||
video_move_responses = {404: VideoNotFoundError("id").openapi}
|
||||
@@ -260,6 +279,7 @@ async def video_delete(
|
||||
|
||||
video_find_responses = {
|
||||
400: SearchPageInvalidError().openapi,
|
||||
401: SearchTokenInvalidError().openapi,
|
||||
404: AlbumNameNotFoundError("name").openapi,
|
||||
422: VideoSearchQueryEmptyError().openapi,
|
||||
}
|
||||
@@ -267,7 +287,7 @@ video_find_responses = {
|
||||
|
||||
@app.get(
|
||||
"/albums/{album}/videos",
|
||||
description="Find a video by filename",
|
||||
description="Find a video by filename, caption or token",
|
||||
response_class=UJSONResponse,
|
||||
response_model=SearchResultsVideo,
|
||||
responses=video_find_responses,
|
||||
@@ -276,10 +296,26 @@ async def video_find(
|
||||
album: str,
|
||||
q: Union[str, None] = None,
|
||||
caption: Union[str, None] = None,
|
||||
token: Union[str, None] = None,
|
||||
page: int = 1,
|
||||
page_size: int = 100,
|
||||
current_user: User = Security(get_current_active_user, scopes=["videos.list"]),
|
||||
):
|
||||
if token is not None:
|
||||
found_record = col_tokens.find_one({"token": token})
|
||||
|
||||
if found_record is None:
|
||||
raise SearchTokenInvalidError()
|
||||
|
||||
return await video_find(
|
||||
album=album,
|
||||
q=found_record["query"],
|
||||
caption=found_record["caption"],
|
||||
page=found_record["page"],
|
||||
page_size=found_record["page_size"],
|
||||
current_user=current_user,
|
||||
)
|
||||
|
||||
if col_albums.find_one({"user": current_user.user, "name": album}) is None:
|
||||
raise AlbumNameNotFoundError(album)
|
||||
|
||||
@@ -341,39 +377,13 @@ async def video_find(
|
||||
{
|
||||
"token": token,
|
||||
"query": q,
|
||||
"album": album,
|
||||
"caption": caption,
|
||||
"page": page + 1,
|
||||
"page_size": page_size,
|
||||
"user": pickle.dumps(current_user),
|
||||
}
|
||||
)
|
||||
output["next_page"] = f"/albums/{album}/videos/token?token={token}" # type: ignore
|
||||
output["next_page"] = f"/albums/{album}/videos/?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"]),
|
||||
)
|
||||
|
@@ -1,8 +1,7 @@
|
||||
from fastapi import FastAPI
|
||||
from fastapi.openapi.docs import get_swagger_ui_html, get_redoc_html
|
||||
from fastapi.openapi.docs import get_redoc_html, get_swagger_ui_html
|
||||
|
||||
|
||||
app = FastAPI(title="END PLAY Photos", docs_url=None, redoc_url=None, version="0.1")
|
||||
app = FastAPI(title="END PLAY Photos", docs_url=None, redoc_url=None, version="0.4")
|
||||
|
||||
|
||||
@app.get("/docs", include_in_schema=False)
|
||||
|
@@ -1,5 +1,6 @@
|
||||
from pymongo import GEOSPHERE, MongoClient
|
||||
|
||||
from modules.utils import configGet
|
||||
from pymongo import MongoClient, GEOSPHERE
|
||||
|
||||
db_config = configGet("database")
|
||||
|
||||
|
@@ -1,8 +1,9 @@
|
||||
from modules.database import col_photos
|
||||
import cv2
|
||||
import numpy as np
|
||||
from numpy.typing import NDArray
|
||||
from scipy import spatial
|
||||
import cv2
|
||||
|
||||
from modules.database import col_photos
|
||||
|
||||
|
||||
def hash_array_to_hash_hex(hash_array):
|
||||
|
@@ -1,6 +1,7 @@
|
||||
from smtplib import SMTP, SMTP_SSL
|
||||
from traceback import print_exc
|
||||
from ssl import create_default_context
|
||||
from traceback import print_exc
|
||||
|
||||
from modules.utils import configGet, logWrite
|
||||
|
||||
try:
|
||||
|
@@ -1,16 +1,13 @@
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from typing import List, Union
|
||||
from modules.database import col_users
|
||||
|
||||
from fastapi import Depends, HTTPException, Security, status
|
||||
from fastapi.security import (
|
||||
OAuth2PasswordBearer,
|
||||
SecurityScopes,
|
||||
)
|
||||
from fastapi.security import OAuth2PasswordBearer, SecurityScopes
|
||||
from jose import JWTError, jwt
|
||||
from passlib.context import CryptContext
|
||||
from pydantic import BaseModel, ValidationError
|
||||
|
||||
from modules.database import col_users
|
||||
|
||||
with open("secret_key", "r", encoding="utf-8") as f:
|
||||
SECRET_KEY = f.read()
|
||||
|
@@ -1,6 +1,7 @@
|
||||
from typing import Any, Union
|
||||
from ujson import loads, dumps, JSONDecodeError
|
||||
from traceback import print_exc
|
||||
from typing import Any, Union
|
||||
|
||||
from ujson import JSONDecodeError, dumps, loads
|
||||
|
||||
|
||||
# Print to stdout and then to log
|
||||
|
@@ -1,10 +1,12 @@
|
||||
from os import makedirs, path
|
||||
from modules.app import app
|
||||
from modules.utils import *
|
||||
from modules.scheduler import scheduler
|
||||
from modules.extensions_loader import dynamic_import_from_src
|
||||
|
||||
from fastapi.responses import FileResponse
|
||||
|
||||
from modules.app import app
|
||||
from modules.extensions_loader import dynamic_import_from_src
|
||||
from modules.scheduler import scheduler
|
||||
from modules.utils import *
|
||||
|
||||
makedirs(path.join("data", "users"), exist_ok=True)
|
||||
|
||||
|
||||
|
@@ -1,10 +1,11 @@
|
||||
fastapi[all]~=0.94.0
|
||||
pymongo==4.3.3
|
||||
ujson~=5.7.0
|
||||
scipy~=1.10.1
|
||||
python-magic~=0.4.27
|
||||
opencv-python~=4.7.0.72
|
||||
python-jose[cryptography]~=3.3.0
|
||||
passlib~=1.7.4
|
||||
aiofiles==23.1.0
|
||||
apscheduler~=3.10.1
|
||||
exif==1.5.0
|
||||
exif==1.6.0
|
||||
fastapi[all]==0.97.0
|
||||
opencv-python~=4.7.0.72
|
||||
passlib~=1.7.4
|
||||
pymongo==4.4.0
|
||||
python-jose[cryptography]~=3.3.0
|
||||
python-magic~=0.4.27
|
||||
scipy~=1.10.1
|
||||
ujson~=5.8.0
|
Reference in New Issue
Block a user