Compare commits
34 Commits
Author | SHA1 | Date | |
---|---|---|---|
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
|
## 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)
|
* [MongoDB](https://www.mongodb.com)
|
||||||
* [exiftool](https://exiftool.org)
|
* [exiftool](https://exiftool.org)
|
||||||
* [jpegoptim](https://github.com/tjko/jpegoptim)
|
* [jpegoptim](https://github.com/tjko/jpegoptim)
|
||||||
|
@@ -1,7 +1,9 @@
|
|||||||
from typing import Literal
|
from typing import Literal
|
||||||
|
|
||||||
|
from fastapi import HTTPException
|
||||||
|
|
||||||
class AlbumNotFoundError(Exception):
|
|
||||||
|
class AlbumNotFoundError(HTTPException):
|
||||||
"""Raises HTTP 404 if no album with this ID found."""
|
"""Raises HTTP 404 if no album with this ID found."""
|
||||||
|
|
||||||
def __init__(self, id: str):
|
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."""
|
"""Raises HTTP 404 if no album with this name found."""
|
||||||
|
|
||||||
def __init__(self, name: str):
|
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."""
|
"""Raises HTTP 409 if album with this name already exists."""
|
||||||
|
|
||||||
def __init__(self, name: str):
|
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."""
|
"""Raises HTTP 406 if album's title or name is invalid."""
|
||||||
|
|
||||||
def __init__(self, place: Literal["name", "title"], error: str) -> None:
|
def __init__(self, place: Literal["name", "title"], error: str) -> None:
|
||||||
@@ -56,13 +70,19 @@ class AlbumIncorrectError(Exception):
|
|||||||
"description": "Album Name/Title Invalid",
|
"description": "Album Name/Title Invalid",
|
||||||
"content": {
|
"content": {
|
||||||
"application/json": {
|
"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."""
|
"""Raises HTTP 404 if no photo with this ID found."""
|
||||||
|
|
||||||
def __init__(self, id: str):
|
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."""
|
"""Raises HTTP 422 if no photo search query provided."""
|
||||||
|
|
||||||
def __init__(self):
|
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."""
|
"""Raises HTTP 404 if no video with this ID found."""
|
||||||
|
|
||||||
def __init__(self, id: str):
|
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."""
|
"""Raises HTTP 422 if no video search query provided."""
|
||||||
|
|
||||||
def __init__(self):
|
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."""
|
"""Raises HTTP 400 if page or page size are not in valid range."""
|
||||||
|
|
||||||
def __init__(self):
|
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."""
|
"""Raises HTTP 401 if search token is not valid."""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -150,9 +194,13 @@ class SearchTokenInvalidError(Exception):
|
|||||||
"application/json": {"example": {"detail": "Invalid search token."}}
|
"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."""
|
"""Raises HTTP 400 if email confirmation code is not valid."""
|
||||||
|
|
||||||
def __init__(self):
|
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."""
|
"""Raises HTTP 409 if user with this name already exists."""
|
||||||
|
|
||||||
def __init__(self):
|
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."""
|
"""Raises HTTP 401 if access token is not valid."""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -190,9 +246,13 @@ class AccessTokenInvalidError(Exception):
|
|||||||
"application/json": {"example": {"detail": "Invalid access token."}}
|
"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."""
|
"""Raises HTTP 401 if user credentials are not valid."""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
@@ -202,3 +262,7 @@ class UserCredentialsInvalid(Exception):
|
|||||||
"application/json": {"example": {"detail": "Invalid credentials."}}
|
"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 typing import List, Union
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
@@ -1,7 +1,5 @@
|
|||||||
from fastapi import Request
|
from fastapi import Request
|
||||||
from fastapi.responses import UJSONResponse
|
from fastapi.responses import UJSONResponse
|
||||||
from modules.app import app
|
|
||||||
from classes.exceptions import *
|
|
||||||
from starlette.status import (
|
from starlette.status import (
|
||||||
HTTP_400_BAD_REQUEST,
|
HTTP_400_BAD_REQUEST,
|
||||||
HTTP_401_UNAUTHORIZED,
|
HTTP_401_UNAUTHORIZED,
|
||||||
@@ -11,6 +9,23 @@ from starlette.status import (
|
|||||||
HTTP_422_UNPROCESSABLE_ENTITY,
|
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)
|
@app.exception_handler(AlbumNotFoundError)
|
||||||
async def album_not_found_exception_handler(request: Request, exc: AlbumNotFoundError):
|
async def album_not_found_exception_handler(request: Request, exc: AlbumNotFoundError):
|
||||||
|
@@ -1,31 +1,42 @@
|
|||||||
from os import path
|
from os import path
|
||||||
from modules.app import app
|
|
||||||
|
import aiofiles
|
||||||
from fastapi.responses import HTMLResponse, Response
|
from fastapi.responses import HTMLResponse, Response
|
||||||
|
|
||||||
|
from modules.app import app
|
||||||
|
|
||||||
|
|
||||||
@app.get("/pages/matter.css", include_in_schema=False)
|
@app.get("/pages/matter.css", include_in_schema=False)
|
||||||
async def page_matter():
|
async def page_matter():
|
||||||
with open(path.join("pages", "matter.css"), "r", encoding="utf-8") as f:
|
async with aiofiles.open(
|
||||||
output = f.read()
|
path.join("pages", "matter.css"), "r", encoding="utf-8"
|
||||||
|
) as f:
|
||||||
|
output = await f.read()
|
||||||
return Response(content=output)
|
return Response(content=output)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/pages/{page}/{file}", include_in_schema=False)
|
@app.get("/pages/{page}/{file}", include_in_schema=False)
|
||||||
async def page_assets(page: str, file: str):
|
async def page_assets(page: str, file: str):
|
||||||
with open(path.join("pages", page, file), "r", encoding="utf-8") as f:
|
async with aiofiles.open(
|
||||||
output = f.read()
|
path.join("pages", page, file), "r", encoding="utf-8"
|
||||||
|
) as f:
|
||||||
|
output = await f.read()
|
||||||
return Response(content=output)
|
return Response(content=output)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/", include_in_schema=False)
|
@app.get("/", include_in_schema=False)
|
||||||
async def page_home():
|
async def page_home():
|
||||||
with open(path.join("pages", "home", "index.html"), "r", encoding="utf-8") as f:
|
async with aiofiles.open(
|
||||||
output = f.read()
|
path.join("pages", "home", "index.html"), "r", encoding="utf-8"
|
||||||
|
) as f:
|
||||||
|
output = await f.read()
|
||||||
return HTMLResponse(content=output)
|
return HTMLResponse(content=output)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/register", include_in_schema=False)
|
@app.get("/register", include_in_schema=False)
|
||||||
async def page_register():
|
async def page_register():
|
||||||
with open(path.join("pages", "register", "index.html"), "r", encoding="utf-8") as f:
|
async with aiofiles.open(
|
||||||
output = f.read()
|
path.join("pages", "register", "index.html"), "r", encoding="utf-8"
|
||||||
|
) as f:
|
||||||
|
output = await f.read()
|
||||||
return HTMLResponse(content=output)
|
return HTMLResponse(content=output)
|
||||||
|
@@ -1,15 +1,24 @@
|
|||||||
import re
|
import re
|
||||||
import pickle
|
from datetime import datetime, timedelta, timezone
|
||||||
|
from os import makedirs, path, remove, system
|
||||||
from secrets import token_urlsafe
|
from secrets import token_urlsafe
|
||||||
from shutil import move
|
from shutil import move
|
||||||
from threading import Thread
|
from threading import Thread
|
||||||
from typing import Union
|
from typing import Union
|
||||||
from uuid import uuid4
|
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 pydantic import ValidationError
|
||||||
|
from pymongo import DESCENDING
|
||||||
|
from starlette.status import HTTP_204_NO_CONTENT, HTTP_409_CONFLICT
|
||||||
|
|
||||||
from classes.exceptions import (
|
from classes.exceptions import (
|
||||||
AccessTokenInvalidError,
|
AccessTokenInvalidError,
|
||||||
AlbumNameNotFoundError,
|
AlbumNameNotFoundError,
|
||||||
@@ -19,8 +28,10 @@ from classes.exceptions import (
|
|||||||
SearchTokenInvalidError,
|
SearchTokenInvalidError,
|
||||||
)
|
)
|
||||||
from classes.models import Photo, PhotoPublic, SearchResultsPhoto
|
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.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.scheduler import scheduler
|
||||||
from modules.security import (
|
from modules.security import (
|
||||||
ALGORITHM,
|
ALGORITHM,
|
||||||
@@ -31,23 +42,6 @@ from modules.security import (
|
|||||||
get_current_active_user,
|
get_current_active_user,
|
||||||
get_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
|
from modules.utils import configGet, logWrite
|
||||||
|
|
||||||
|
|
||||||
@@ -130,7 +124,7 @@ async def photo_upload(
|
|||||||
".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
|
".".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"
|
path.join("data", "users", current_user.user, "albums", album, filename), "wb"
|
||||||
) as f:
|
) as f:
|
||||||
f.write(await file.read())
|
f.write(await file.read())
|
||||||
@@ -266,16 +260,24 @@ if configGet("media_token_access") is True:
|
|||||||
|
|
||||||
mime = Magic(mime=True).from_file(image_path)
|
mime = Magic(mime=True).from_file(image_path)
|
||||||
|
|
||||||
with open(image_path, "rb") as f:
|
async with aiofiles.open(image_path, "rb") as f:
|
||||||
image_file = f.read()
|
image_file = await f.read()
|
||||||
|
|
||||||
return Response(image_file, media_type=mime)
|
return Response(image_file, media_type=mime)
|
||||||
|
|
||||||
|
|
||||||
photo_get_responses = {404: PhotoNotFoundError("id").openapi}
|
photo_get_responses = {
|
||||||
|
200: {"content": {"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(
|
async def photo_get(
|
||||||
id: str,
|
id: str,
|
||||||
current_user: User = Security(get_current_active_user, scopes=["photos.read"]),
|
current_user: User = Security(get_current_active_user, scopes=["photos.read"]),
|
||||||
@@ -293,8 +295,8 @@ async def photo_get(
|
|||||||
|
|
||||||
mime = Magic(mime=True).from_file(image_path)
|
mime = Magic(mime=True).from_file(image_path)
|
||||||
|
|
||||||
with open(image_path, "rb") as f:
|
async with aiofiles.open(image_path, "rb") as f:
|
||||||
image_file = f.read()
|
image_file = await f.read()
|
||||||
|
|
||||||
return Response(image_file, media_type=mime)
|
return Response(image_file, media_type=mime)
|
||||||
|
|
||||||
@@ -444,6 +446,7 @@ async def photo_delete(
|
|||||||
|
|
||||||
photo_find_responses = {
|
photo_find_responses = {
|
||||||
400: SearchPageInvalidError().openapi,
|
400: SearchPageInvalidError().openapi,
|
||||||
|
401: SearchTokenInvalidError().openapi,
|
||||||
404: AlbumNameNotFoundError("name").openapi,
|
404: AlbumNameNotFoundError("name").openapi,
|
||||||
422: PhotoSearchQueryEmptyError().openapi,
|
422: PhotoSearchQueryEmptyError().openapi,
|
||||||
}
|
}
|
||||||
@@ -451,7 +454,7 @@ photo_find_responses = {
|
|||||||
|
|
||||||
@app.get(
|
@app.get(
|
||||||
"/albums/{album}/photos",
|
"/albums/{album}/photos",
|
||||||
description="Find a photo by filename",
|
description="Find a photo by filename, caption, location or token",
|
||||||
response_class=UJSONResponse,
|
response_class=UJSONResponse,
|
||||||
response_model=SearchResultsPhoto,
|
response_model=SearchResultsPhoto,
|
||||||
responses=photo_find_responses,
|
responses=photo_find_responses,
|
||||||
@@ -460,6 +463,7 @@ async def photo_find(
|
|||||||
album: str,
|
album: str,
|
||||||
q: Union[str, None] = None,
|
q: Union[str, None] = None,
|
||||||
caption: Union[str, None] = None,
|
caption: Union[str, None] = None,
|
||||||
|
token: Union[str, None] = None,
|
||||||
page: int = 1,
|
page: int = 1,
|
||||||
page_size: int = 100,
|
page_size: int = 100,
|
||||||
lat: Union[float, None] = None,
|
lat: Union[float, None] = None,
|
||||||
@@ -467,6 +471,24 @@ async def photo_find(
|
|||||||
radius: Union[int, None] = None,
|
radius: Union[int, None] = None,
|
||||||
current_user: User = Security(get_current_active_user, scopes=["photos.list"]),
|
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:
|
if col_albums.find_one({"user": current_user.user, "name": album}) is None:
|
||||||
raise AlbumNameNotFoundError(album)
|
raise AlbumNameNotFoundError(album)
|
||||||
|
|
||||||
@@ -543,39 +565,16 @@ async def photo_find(
|
|||||||
{
|
{
|
||||||
"token": token,
|
"token": token,
|
||||||
"query": q,
|
"query": q,
|
||||||
"album": album,
|
"caption": caption,
|
||||||
|
"lat": lat,
|
||||||
|
"lng": lng,
|
||||||
|
"radius": radius,
|
||||||
"page": page + 1,
|
"page": page + 1,
|
||||||
"page_size": page_size,
|
"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:
|
else:
|
||||||
output["next_page"] = None # type: ignore
|
output["next_page"] = None # type: ignore
|
||||||
|
|
||||||
return UJSONResponse(output)
|
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 datetime import timedelta
|
||||||
from classes.exceptions import UserCredentialsInvalid
|
|
||||||
from modules.app import app
|
|
||||||
|
|
||||||
from fastapi import Depends
|
from fastapi import Depends
|
||||||
from fastapi.security import (
|
from fastapi.security import OAuth2PasswordRequestForm
|
||||||
OAuth2PasswordRequestForm,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
from classes.exceptions import UserCredentialsInvalid
|
||||||
|
from modules.app import app
|
||||||
from modules.security import (
|
from modules.security import (
|
||||||
ACCESS_TOKEN_EXPIRE_DAYS,
|
ACCESS_TOKEN_EXPIRE_DAYS,
|
||||||
Token,
|
Token,
|
||||||
|
@@ -1,27 +1,19 @@
|
|||||||
from datetime import datetime, timedelta
|
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 (
|
from classes.exceptions import (
|
||||||
UserAlreadyExists,
|
UserAlreadyExists,
|
||||||
UserCredentialsInvalid,
|
UserCredentialsInvalid,
|
||||||
UserEmailCodeInvalid,
|
UserEmailCodeInvalid,
|
||||||
)
|
)
|
||||||
from modules.database import (
|
|
||||||
col_users,
|
|
||||||
col_albums,
|
|
||||||
col_photos,
|
|
||||||
col_emails,
|
|
||||||
col_videos,
|
|
||||||
col_emails,
|
|
||||||
)
|
|
||||||
from modules.app import app
|
from modules.app import app
|
||||||
from modules.utils import configGet, logWrite
|
from modules.database import col_albums, col_emails, col_photos, col_users, col_videos
|
||||||
from modules.scheduler import scheduler
|
|
||||||
from modules.mailer import mail_sender
|
from modules.mailer import mail_sender
|
||||||
|
from modules.scheduler import scheduler
|
||||||
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.security import (
|
from modules.security import (
|
||||||
User,
|
User,
|
||||||
get_current_active_user,
|
get_current_active_user,
|
||||||
@@ -29,6 +21,7 @@ from modules.security import (
|
|||||||
get_user,
|
get_user,
|
||||||
verify_password,
|
verify_password,
|
||||||
)
|
)
|
||||||
|
from modules.utils import configGet, logWrite
|
||||||
|
|
||||||
|
|
||||||
async def send_confirmation(user: str, email: str):
|
async def send_confirmation(user: str, email: str):
|
||||||
|
@@ -1,11 +1,19 @@
|
|||||||
import re
|
import re
|
||||||
import pickle
|
from datetime import datetime, timezone
|
||||||
|
from os import makedirs, path, remove
|
||||||
from secrets import token_urlsafe
|
from secrets import token_urlsafe
|
||||||
from shutil import move
|
from shutil import move
|
||||||
from typing import Union
|
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 magic import Magic
|
||||||
from datetime import datetime, timezone
|
from pymongo import DESCENDING
|
||||||
from os import makedirs, path, remove
|
from starlette.status import HTTP_204_NO_CONTENT
|
||||||
|
|
||||||
from classes.exceptions import (
|
from classes.exceptions import (
|
||||||
AlbumNameNotFoundError,
|
AlbumNameNotFoundError,
|
||||||
SearchPageInvalidError,
|
SearchPageInvalidError,
|
||||||
@@ -13,17 +21,10 @@ from classes.exceptions import (
|
|||||||
VideoNotFoundError,
|
VideoNotFoundError,
|
||||||
VideoSearchQueryEmptyError,
|
VideoSearchQueryEmptyError,
|
||||||
)
|
)
|
||||||
from classes.models import Video, SearchResultsVideo, VideoPublic
|
from classes.models import SearchResultsVideo, Video, VideoPublic
|
||||||
from modules.security import User, get_current_active_user
|
|
||||||
from modules.app import app
|
from modules.app import app
|
||||||
from modules.database import col_videos, col_albums, col_tokens
|
from modules.database import col_albums, col_tokens, col_videos
|
||||||
from bson.objectid import ObjectId
|
from modules.security import User, get_current_active_user
|
||||||
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}
|
video_post_responses = {404: AlbumNameNotFoundError("name").openapi}
|
||||||
|
|
||||||
@@ -59,10 +60,10 @@ async def video_upload(
|
|||||||
".".join(base_name) + f"_{int(datetime.now().timestamp())}." + extension
|
".".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"
|
path.join("data", "users", current_user.user, "albums", album, filename), "wb"
|
||||||
) as f:
|
) as f:
|
||||||
f.write(await file.read())
|
await f.write(await file.read())
|
||||||
|
|
||||||
# Hashing and duplicates check should be here
|
# Hashing and duplicates check should be here
|
||||||
|
|
||||||
@@ -91,10 +92,18 @@ async def video_upload(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
video_get_responses = {404: VideoNotFoundError("id").openapi}
|
video_get_responses = {
|
||||||
|
200: {"content": {"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(
|
async def video_get(
|
||||||
id: str,
|
id: str,
|
||||||
current_user: User = Security(get_current_active_user, scopes=["videos.read"]),
|
current_user: User = Security(get_current_active_user, scopes=["videos.read"]),
|
||||||
@@ -112,10 +121,10 @@ async def video_get(
|
|||||||
|
|
||||||
mime = Magic(mime=True).from_file(video_path)
|
mime = Magic(mime=True).from_file(video_path)
|
||||||
|
|
||||||
with open(video_path, "rb") as f:
|
async with aiofiles.open(video_path, "rb") as f:
|
||||||
video_file = f.read()
|
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}
|
video_move_responses = {404: VideoNotFoundError("id").openapi}
|
||||||
@@ -260,6 +269,7 @@ async def video_delete(
|
|||||||
|
|
||||||
video_find_responses = {
|
video_find_responses = {
|
||||||
400: SearchPageInvalidError().openapi,
|
400: SearchPageInvalidError().openapi,
|
||||||
|
401: SearchTokenInvalidError().openapi,
|
||||||
404: AlbumNameNotFoundError("name").openapi,
|
404: AlbumNameNotFoundError("name").openapi,
|
||||||
422: VideoSearchQueryEmptyError().openapi,
|
422: VideoSearchQueryEmptyError().openapi,
|
||||||
}
|
}
|
||||||
@@ -267,7 +277,7 @@ video_find_responses = {
|
|||||||
|
|
||||||
@app.get(
|
@app.get(
|
||||||
"/albums/{album}/videos",
|
"/albums/{album}/videos",
|
||||||
description="Find a video by filename",
|
description="Find a video by filename, caption or token",
|
||||||
response_class=UJSONResponse,
|
response_class=UJSONResponse,
|
||||||
response_model=SearchResultsVideo,
|
response_model=SearchResultsVideo,
|
||||||
responses=video_find_responses,
|
responses=video_find_responses,
|
||||||
@@ -276,10 +286,26 @@ async def video_find(
|
|||||||
album: str,
|
album: str,
|
||||||
q: Union[str, None] = None,
|
q: Union[str, None] = None,
|
||||||
caption: Union[str, None] = None,
|
caption: Union[str, None] = None,
|
||||||
|
token: Union[str, None] = None,
|
||||||
page: int = 1,
|
page: int = 1,
|
||||||
page_size: int = 100,
|
page_size: int = 100,
|
||||||
current_user: User = Security(get_current_active_user, scopes=["videos.list"]),
|
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:
|
if col_albums.find_one({"user": current_user.user, "name": album}) is None:
|
||||||
raise AlbumNameNotFoundError(album)
|
raise AlbumNameNotFoundError(album)
|
||||||
|
|
||||||
@@ -341,39 +367,13 @@ async def video_find(
|
|||||||
{
|
{
|
||||||
"token": token,
|
"token": token,
|
||||||
"query": q,
|
"query": q,
|
||||||
"album": album,
|
"caption": caption,
|
||||||
"page": page + 1,
|
"page": page + 1,
|
||||||
"page_size": page_size,
|
"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:
|
else:
|
||||||
output["next_page"] = None # type: ignore
|
output["next_page"] = None # type: ignore
|
||||||
|
|
||||||
return UJSONResponse(output)
|
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 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.3")
|
||||||
app = FastAPI(title="END PLAY Photos", docs_url=None, redoc_url=None, version="0.1")
|
|
||||||
|
|
||||||
|
|
||||||
@app.get("/docs", include_in_schema=False)
|
@app.get("/docs", include_in_schema=False)
|
||||||
|
@@ -1,5 +1,6 @@
|
|||||||
|
from pymongo import GEOSPHERE, MongoClient
|
||||||
|
|
||||||
from modules.utils import configGet
|
from modules.utils import configGet
|
||||||
from pymongo import MongoClient, GEOSPHERE
|
|
||||||
|
|
||||||
db_config = configGet("database")
|
db_config = configGet("database")
|
||||||
|
|
||||||
|
@@ -1,8 +1,9 @@
|
|||||||
from modules.database import col_photos
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from numpy.typing import NDArray
|
from numpy.typing import NDArray
|
||||||
from scipy import spatial
|
from scipy import spatial
|
||||||
import cv2
|
|
||||||
|
from modules.database import col_photos
|
||||||
|
|
||||||
|
|
||||||
def hash_array_to_hash_hex(hash_array):
|
def hash_array_to_hash_hex(hash_array):
|
||||||
|
@@ -1,6 +1,7 @@
|
|||||||
from smtplib import SMTP, SMTP_SSL
|
from smtplib import SMTP, SMTP_SSL
|
||||||
from traceback import print_exc
|
|
||||||
from ssl import create_default_context
|
from ssl import create_default_context
|
||||||
|
from traceback import print_exc
|
||||||
|
|
||||||
from modules.utils import configGet, logWrite
|
from modules.utils import configGet, logWrite
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@@ -1,16 +1,13 @@
|
|||||||
from datetime import datetime, timedelta, timezone
|
from datetime import datetime, timedelta, timezone
|
||||||
from typing import List, Union
|
from typing import List, Union
|
||||||
from modules.database import col_users
|
|
||||||
|
|
||||||
from fastapi import Depends, HTTPException, Security, status
|
from fastapi import Depends, HTTPException, Security, status
|
||||||
from fastapi.security import (
|
from fastapi.security import OAuth2PasswordBearer, SecurityScopes
|
||||||
OAuth2PasswordBearer,
|
|
||||||
SecurityScopes,
|
|
||||||
)
|
|
||||||
from jose import JWTError, jwt
|
from jose import JWTError, jwt
|
||||||
from passlib.context import CryptContext
|
from passlib.context import CryptContext
|
||||||
from pydantic import BaseModel, ValidationError
|
from pydantic import BaseModel, ValidationError
|
||||||
|
|
||||||
|
from modules.database import col_users
|
||||||
|
|
||||||
with open("secret_key", "r", encoding="utf-8") as f:
|
with open("secret_key", "r", encoding="utf-8") as f:
|
||||||
SECRET_KEY = f.read()
|
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 traceback import print_exc
|
||||||
|
from typing import Any, Union
|
||||||
|
|
||||||
|
from ujson import JSONDecodeError, dumps, loads
|
||||||
|
|
||||||
|
|
||||||
# Print to stdout and then to log
|
# Print to stdout and then to log
|
||||||
|
@@ -1,10 +1,12 @@
|
|||||||
from os import makedirs, path
|
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 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)
|
makedirs(path.join("data", "users"), exist_ok=True)
|
||||||
|
|
||||||
|
|
||||||
|
@@ -1,10 +1,11 @@
|
|||||||
fastapi[all]~=0.94.0
|
aiofiles==23.1.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
|
|
||||||
apscheduler~=3.10.1
|
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