first
This commit is contained in:
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@@ -0,0 +1,2 @@
|
||||
*.cpython
|
||||
__pycache__
|
||||
1
Dockerfile
Normal file
1
Dockerfile
Normal file
@@ -0,0 +1 @@
|
||||
FROM python:3.11.3
|
||||
89
alembic.ini
Normal file
89
alembic.ini
Normal file
@@ -0,0 +1,89 @@
|
||||
# A generic, single database configuration.
|
||||
|
||||
[alembic]
|
||||
# path to migration scripts
|
||||
script_location = migration
|
||||
|
||||
# template used to generate migration files
|
||||
# file_template = %%(rev)s_%%(slug)s
|
||||
|
||||
# sys.path path, will be prepended to sys.path if present.
|
||||
# defaults to the current working directory.
|
||||
prepend_sys_path = .
|
||||
|
||||
# timezone to use when rendering the date
|
||||
# within the migration file as well as the filename.
|
||||
# string value is passed to dateutil.tz.gettz()
|
||||
# leave blank for localtime
|
||||
# timezone =
|
||||
|
||||
# max length of characters to apply to the
|
||||
# "slug" field
|
||||
# truncate_slug_length = 40
|
||||
|
||||
# set to 'true' to run the environment during
|
||||
# the 'revision' command, regardless of autogenerate
|
||||
# revision_environment = false
|
||||
|
||||
# set to 'true' to allow .pyc and .pyo files without
|
||||
# a source .py file to be detected as revisions in the
|
||||
# versions/ directory
|
||||
# sourceless = false
|
||||
|
||||
# version location specification; this defaults
|
||||
# to migration/versions. When using multiple version
|
||||
# directories, initial revisions must be specified with --version-path
|
||||
# version_locations = %(here)s/bar %(here)s/bat migration/versions
|
||||
|
||||
# the output encoding used when revision files
|
||||
# are written from script.py.mako
|
||||
# output_encoding = utf-8
|
||||
|
||||
sqlalchemy.url = postgresql://barbie:redStar4,1@localhost:5432/personal_info_db
|
||||
|
||||
|
||||
[post_write_hooks]
|
||||
# post_write_hooks defines scripts or Python functions that are run
|
||||
# on newly generated revision scripts. See the documentation for further
|
||||
# detail and examples
|
||||
|
||||
# format using "black" - use the console_scripts runner, against the "black" entrypoint
|
||||
# hooks = black
|
||||
# black.type = console_scripts
|
||||
# black.entrypoint = black
|
||||
# black.options = -l 79 REVISION_SCRIPT_FILENAME
|
||||
|
||||
# Logging configuration
|
||||
[loggers]
|
||||
keys = root,sqlalchemy,alembic
|
||||
|
||||
[handlers]
|
||||
keys = console
|
||||
|
||||
[formatters]
|
||||
keys = generic
|
||||
|
||||
[logger_root]
|
||||
level = WARN
|
||||
handlers = console
|
||||
qualname =
|
||||
|
||||
[logger_sqlalchemy]
|
||||
level = WARN
|
||||
handlers =
|
||||
qualname = sqlalchemy.engine
|
||||
|
||||
[logger_alembic]
|
||||
level = INFO
|
||||
handlers =
|
||||
qualname = alembic
|
||||
|
||||
[handler_console]
|
||||
class = StreamHandler
|
||||
args = (sys.stderr,)
|
||||
level = NOTSET
|
||||
formatter = generic
|
||||
|
||||
[formatter_generic]
|
||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
||||
datefmt = %H:%M:%S
|
||||
1
migration/README
Normal file
1
migration/README
Normal file
@@ -0,0 +1 @@
|
||||
Generic single-database configuration.
|
||||
78
migration/env.py
Normal file
78
migration/env.py
Normal file
@@ -0,0 +1,78 @@
|
||||
from logging.config import fileConfig
|
||||
|
||||
from sqlalchemy import engine_from_config
|
||||
from sqlalchemy import pool
|
||||
|
||||
from alembic import context
|
||||
from personal_info.db.models import Base
|
||||
|
||||
# this is the Alembic Config object, which provides
|
||||
# access to the values within the .ini file in use.
|
||||
config = context.config
|
||||
|
||||
# Interpret the config file for Python logging.
|
||||
# This line sets up loggers basically.
|
||||
fileConfig(config.config_file_name)
|
||||
|
||||
# add your model's MetaData object here
|
||||
# for 'autogenerate' support
|
||||
# from myapp import mymodel
|
||||
# target_metadata = mymodel.Base.metadata
|
||||
target_metadata = Base.metadata
|
||||
|
||||
# other values from the config, defined by the needs of env.py,
|
||||
# can be acquired:
|
||||
# my_important_option = config.get_main_option("my_important_option")
|
||||
# ... etc.
|
||||
|
||||
|
||||
def run_migrations_offline():
|
||||
"""Run migrations in 'offline' mode.
|
||||
|
||||
This configures the context with just a URL
|
||||
and not an Engine, though an Engine is acceptable
|
||||
here as well. By skipping the Engine creation
|
||||
we don't even need a DBAPI to be available.
|
||||
|
||||
Calls to context.execute() here emit the given string to the
|
||||
script output.
|
||||
|
||||
"""
|
||||
url = config.get_main_option("sqlalchemy.url")
|
||||
context.configure(
|
||||
url=url,
|
||||
target_metadata=target_metadata,
|
||||
literal_binds=True,
|
||||
dialect_opts={"paramstyle": "named"},
|
||||
)
|
||||
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
def run_migrations_online():
|
||||
"""Run migrations in 'online' mode.
|
||||
|
||||
In this scenario we need to create an Engine
|
||||
and associate a connection with the context.
|
||||
|
||||
"""
|
||||
connectable = engine_from_config(
|
||||
config.get_section(config.config_ini_section),
|
||||
prefix="sqlalchemy.",
|
||||
poolclass=pool.NullPool,
|
||||
)
|
||||
|
||||
with connectable.connect() as connection:
|
||||
context.configure(
|
||||
connection=connection, target_metadata=target_metadata
|
||||
)
|
||||
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
if context.is_offline_mode():
|
||||
run_migrations_offline()
|
||||
else:
|
||||
run_migrations_online()
|
||||
24
migration/script.py.mako
Normal file
24
migration/script.py.mako
Normal file
@@ -0,0 +1,24 @@
|
||||
"""${message}
|
||||
|
||||
Revision ID: ${up_revision}
|
||||
Revises: ${down_revision | comma,n}
|
||||
Create Date: ${create_date}
|
||||
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
${imports if imports else ""}
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = ${repr(up_revision)}
|
||||
down_revision = ${repr(down_revision)}
|
||||
branch_labels = ${repr(branch_labels)}
|
||||
depends_on = ${repr(depends_on)}
|
||||
|
||||
|
||||
def upgrade():
|
||||
${upgrades if upgrades else "pass"}
|
||||
|
||||
|
||||
def downgrade():
|
||||
${downgrades if downgrades else "pass"}
|
||||
5
notes.txt
Normal file
5
notes.txt
Normal file
@@ -0,0 +1,5 @@
|
||||
export PYTHONPATH='/home/dd/astra/personal_information/src'
|
||||
docker run -itd -e POSTGRES_USER=barbie -e POSTGRES_PASSWORD=redStar4,1 -p 5432:5432 -v /home/dd/astra/postgresql_db:/var/lib/postgresql/data --name personal_info_db postgres
|
||||
docker exec -it personal_info_db bash
|
||||
psql -U barbie -d barbie
|
||||
CREATE DATABASE personal_info_db;
|
||||
67
requirements.txt
Normal file
67
requirements.txt
Normal file
@@ -0,0 +1,67 @@
|
||||
# This file may be used to create an environment using:
|
||||
# $ conda create --name <env> --file <this file>
|
||||
# platform: linux-64
|
||||
_libgcc_mutex=0.1=conda_forge
|
||||
_openmp_mutex=4.5=2_gnu
|
||||
alembic=1.10.4=pyhd8ed1ab_0
|
||||
anyio=3.6.2=pyhd8ed1ab_0
|
||||
bzip2=1.0.8=h7f98852_4
|
||||
ca-certificates=2022.12.7=ha878542_0
|
||||
certifi=2022.12.7=pyhd8ed1ab_0
|
||||
click=8.1.3=py311h38be061_1
|
||||
colorama=0.4.6=pyhd8ed1ab_0
|
||||
exceptiongroup=1.1.1=pyhd8ed1ab_0
|
||||
fastapi=0.95.1=pyhd8ed1ab_0
|
||||
greenlet=2.0.2=py311hcafe171_0
|
||||
h11=0.14.0=pyhd8ed1ab_0
|
||||
h2=4.1.0=pyhd8ed1ab_0
|
||||
hpack=4.0.0=pyh9f0ad1d_0
|
||||
httpcore=0.17.0=pyhd8ed1ab_0
|
||||
httpx=0.24.0=pyhd8ed1ab_1
|
||||
hyperframe=6.0.1=pyhd8ed1ab_0
|
||||
idna=3.4=pyhd8ed1ab_0
|
||||
importlib-metadata=6.6.0=pyha770c72_0
|
||||
importlib_resources=5.12.0=pyhd8ed1ab_0
|
||||
iniconfig=2.0.0=pyhd8ed1ab_0
|
||||
keyutils=1.6.1=h166bdaf_0
|
||||
krb5=1.20.1=h81ceb04_0
|
||||
ld_impl_linux-64=2.40=h41732ed_0
|
||||
libedit=3.1.20191231=he28a2e2_2
|
||||
libexpat=2.5.0=hcb278e6_1
|
||||
libffi=3.4.2=h7f98852_5
|
||||
libgcc-ng=12.2.0=h65d4601_19
|
||||
libgomp=12.2.0=h65d4601_19
|
||||
libnsl=2.0.0=h7f98852_0
|
||||
libpq=15.2=hb675445_0
|
||||
libsqlite=3.40.0=h753d276_1
|
||||
libstdcxx-ng=12.2.0=h46fd767_19
|
||||
libuuid=2.38.1=h0b41bf4_0
|
||||
libzlib=1.2.13=h166bdaf_4
|
||||
mako=1.2.4=pyhd8ed1ab_0
|
||||
markupsafe=2.1.2=py311h2582759_0
|
||||
ncurses=6.3=h27087fc_1
|
||||
openssl=3.1.0=hd590300_3
|
||||
packaging=23.1=pyhd8ed1ab_0
|
||||
pip=23.1.2=pyhd8ed1ab_0
|
||||
pluggy=1.0.0=py311h38be061_4
|
||||
psycopg2=2.9.3=py311h968e94b_2
|
||||
psycopg2-binary=2.9.3=pyhd8ed1ab_2
|
||||
pydantic=1.10.7=py311h2582759_0
|
||||
pytest=7.3.1=pyhd8ed1ab_0
|
||||
python=3.11.3=h2755cc3_0_cpython
|
||||
python-dotenv=1.0.0=pyhd8ed1ab_0
|
||||
python_abi=3.11=3_cp311
|
||||
readline=8.2=h8228510_1
|
||||
setuptools=67.7.2=pyhd8ed1ab_0
|
||||
sniffio=1.3.0=pyhd8ed1ab_0
|
||||
sqlalchemy=2.0.11=py311h459d7ec_0
|
||||
starlette=0.26.1=pyhd8ed1ab_0
|
||||
tk=8.6.12=h27826a3_0
|
||||
tomli=2.0.1=pyhd8ed1ab_0
|
||||
typing-extensions=4.5.0=hd8ed1ab_0
|
||||
typing_extensions=4.5.0=pyha770c72_0
|
||||
tzdata=2023c=h71feb2d_0
|
||||
uvicorn=0.22.0=py311h38be061_0
|
||||
wheel=0.40.0=pyhd8ed1ab_0
|
||||
xz=5.2.6=h166bdaf_0
|
||||
zipp=3.15.0=pyhd8ed1ab_0
|
||||
13
setup.py
Normal file
13
setup.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from setuptools import find_packages, setup
|
||||
|
||||
install_requires = ['fastapi', 'uvicorn']
|
||||
|
||||
setup(
|
||||
name='personal_info',
|
||||
install_requires=install_requires,
|
||||
use_scm_version=True,
|
||||
setup_requires=['setuptools_scm'],
|
||||
packages=find_packages(where='src'),
|
||||
package_dir={'':'src'},
|
||||
include_package_data=True,
|
||||
)
|
||||
0
src/personal_info/__init__.py
Normal file
0
src/personal_info/__init__.py
Normal file
19
src/personal_info/app.py
Normal file
19
src/personal_info/app.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from fastapi import FastAPI
|
||||
|
||||
def create_app() -> FastAPI:
|
||||
app = FastAPI (
|
||||
title='personal_info',
|
||||
description='Personal Information Service',
|
||||
version='1.0'
|
||||
)
|
||||
|
||||
@app.get('/health')
|
||||
async def health() -> str:
|
||||
return 'ok'
|
||||
|
||||
|
||||
from personal_info.routers import persons
|
||||
|
||||
app.include(persons.router)
|
||||
|
||||
return app
|
||||
15
src/personal_info/config.py
Normal file
15
src/personal_info/config.py
Normal file
@@ -0,0 +1,15 @@
|
||||
from functools import lru_cache
|
||||
from typing import Optional
|
||||
|
||||
from pydantic import BaseSettings, PostgresDsn
|
||||
|
||||
class Settings(BaseSettings):
|
||||
database_url: Optional[PostgresDsn]
|
||||
|
||||
@lru_cache
|
||||
def get_settings() -> Settings:
|
||||
settings = Settings() # type: ignore
|
||||
print('333333333333333333333333333333333333333333')
|
||||
print(settings)
|
||||
#CONTINUE
|
||||
return settings
|
||||
0
src/personal_info/db/__init__.py
Normal file
0
src/personal_info/db/__init__.py
Normal file
17
src/personal_info/db/models.py
Normal file
17
src/personal_info/db/models.py
Normal file
@@ -0,0 +1,17 @@
|
||||
import uuid
|
||||
from typing import Any
|
||||
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
from sqlalchemy.orm import relationship
|
||||
from sqlalchemy.sql.schema import ForeignKey
|
||||
|
||||
Base: Any = declarative_base()
|
||||
|
||||
class Person(Base):
|
||||
__tablename__ = 'people'
|
||||
|
||||
id = sa.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
first_name = sa.Column(sa.Text, nullable=False)
|
||||
last_name = sa.Column(sa.Text, nullable=False)
|
||||
1
src/personal_info/db/schemas/__init__.py
Normal file
1
src/personal_info/db/schemas/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
from .person import *
|
||||
16
src/personal_info/db/schemas/person.py
Normal file
16
src/personal_info/db/schemas/person.py
Normal file
@@ -0,0 +1,16 @@
|
||||
from datetime import datetime
|
||||
from typing import List
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
from pydantic.types import UUID4
|
||||
|
||||
class Person(BaseModel):
|
||||
id: UUID4
|
||||
first_name: str
|
||||
last_name: str
|
||||
|
||||
class PersonCreate(Person):
|
||||
pass
|
||||
|
||||
class PersonDetail(Person):
|
||||
pass
|
||||
23
src/personal_info/db/session.py
Normal file
23
src/personal_info/db/session.py
Normal file
@@ -0,0 +1,23 @@
|
||||
from functools import lru_cache
|
||||
from typing import Generator
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import scoped_session, sessionmaker
|
||||
|
||||
from personal_info.config import get_settings
|
||||
|
||||
engine = create_engine(get_settings().database_url, pool_pre_ping=True)
|
||||
|
||||
@lru_cache
|
||||
def create_session() -> scoped_session:
|
||||
Session = scoped_session(
|
||||
sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
)
|
||||
return Session
|
||||
|
||||
def get_session() -> Generator[scoped_session, None, None]:
|
||||
Session = create_session()
|
||||
try:
|
||||
yield Session
|
||||
finally:
|
||||
Session.remove()
|
||||
3
src/personal_info/main.py
Normal file
3
src/personal_info/main.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from personal_info.app import create_app
|
||||
|
||||
app = create_app()
|
||||
0
src/personal_info/routers/__init__.py
Normal file
0
src/personal_info/routers/__init__.py
Normal file
27
src/personal_info/routers/persons.py
Normal file
27
src/personal_info/routers/persons.py
Normal file
@@ -0,0 +1,27 @@
|
||||
from fastapi import Depends
|
||||
from fastapi.routing import APIRouter
|
||||
from pydantic.types import UUID4
|
||||
|
||||
from personal_info.db import models
|
||||
from personal_info.db.schemas import Person, PersonDetail
|
||||
from personal_info.services import PersonService, get_persons_service
|
||||
|
||||
router = APIrouter(prefix='/people')
|
||||
|
||||
@router.get('/', response_model=List[Person])
|
||||
async def list_persons(
|
||||
persons_service: PersonService = Depends(get_persons_service),
|
||||
) -> List[models.Person]:
|
||||
return persons_service.list()
|
||||
|
||||
@router.get('/{person_id}', response_model=PersonDetail)
|
||||
async def get_person(
|
||||
person_id: UUID4, persons_service: PersonService = Depends(get_persons_service)
|
||||
) -> Optional[models.Person]:
|
||||
return persons_service.get(person_id)
|
||||
|
||||
@router.post('/', status_code=201, response_model=PersonDetail)
|
||||
async def create_person(
|
||||
person: PersonCreate, persons_service: PersonService = Depends(get_persons_service)
|
||||
) -> models.Person:
|
||||
return persons_service.create(person)
|
||||
11
src/personal_info/services/__init__.py
Normal file
11
src/personal_info/services/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from fastapi import Depends
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from personal_info.db.session import get_session
|
||||
|
||||
from .people import PersonService
|
||||
|
||||
def get_persons_service(db_session: Session = Depends(get_session)) -> PersonService:
|
||||
return PersonService(db_session)
|
||||
|
||||
__all__ = ('get_persons_service')
|
||||
52
src/personal_info/services/base.py
Normal file
52
src/personal_info/services/base.py
Normal file
@@ -0,0 +1,52 @@
|
||||
from typing import Any, Generic, List, Optional, Type, TypeVar
|
||||
|
||||
import sqlalchemy
|
||||
from pydantic import BaseModel
|
||||
from sqlalchemy.orm import Session
|
||||
from starlette.exceptions import HTTPException
|
||||
|
||||
from personal_info.db.models import Base
|
||||
|
||||
ModelType = TypeVar('ModelType', bound=Base)
|
||||
CreateSchemaType = TypeVar('CreateSchemaType', bound=BaseModel)
|
||||
UpdateSchemaType = TypeVar('UpdateSchemaType', bound=BaseModel)
|
||||
|
||||
class BaseService(Generic[ModelType, CreateSchemaType, UpdateSchemaType]):
|
||||
def __init__(self, model: Type[ModelType], db_session: Session):
|
||||
self.model = model
|
||||
self.db_session = db_session
|
||||
|
||||
def get(self, id: Any) -> Optional[ModelType]:
|
||||
obj: Optional[ModelType] = self.db_session.get(self.model, id)
|
||||
if obj is None:
|
||||
raise HTTPException(status_code=404, detail='NotFound')
|
||||
return obj
|
||||
|
||||
def list(self) -> List[ModelType]:
|
||||
objs: List[ModelType] = self.db_session.query(self.model).all()
|
||||
return objs
|
||||
|
||||
def create(self, obj: CreateSchemaType) -> ModelType:
|
||||
db_obj: ModelType = self.model(**obj.dict())
|
||||
self.db_session.add(db_obj)
|
||||
try:
|
||||
self.db_session.commit()
|
||||
except sqlalchemy.exc.IntegrityError as e:
|
||||
self.db_session.rollback()
|
||||
if 'duplicate key' in str(e):
|
||||
raise HTTPException(status_code=409, detail='Conflict Error')
|
||||
else:
|
||||
raise e
|
||||
return db_obj
|
||||
|
||||
def update(self, id: Any, obj: UpdateSchemaType) -> Optional[ModelType]:
|
||||
db_obj = self.get(id)
|
||||
for column, value in obj.dict(exclude_unset=True).items():
|
||||
setattr(db_obj, column, value)
|
||||
self.db_session.commit()
|
||||
return db_obj
|
||||
|
||||
def delete(self, id: Any) -> None:
|
||||
db_obj = self.db_session.get(self.model, id)
|
||||
self.db_session.delete(db_obj)
|
||||
self.db_session.commit()
|
||||
19
src/personal_info/services/people.py
Normal file
19
src/personal_info/services/people.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from typing import Any
|
||||
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from personal_info.db.models import Person
|
||||
from personal_info.db.schemas import PersonCreate
|
||||
|
||||
from personal_info.services.base import BaseService
|
||||
|
||||
class PersonService(BaseService[Person, PersonCreate, Any]):
|
||||
def __init__(self, db_session: Session):
|
||||
super(PersonService, self).__init__(Person, db_session)
|
||||
|
||||
def create(self, obj: PersonCreate) -> Person:
|
||||
person: Person = Person()
|
||||
self.db_session.add(person)
|
||||
self.db_session.flush()
|
||||
self.db_session.commit()
|
||||
return person
|
||||
8
tests/conftest.py
Normal file
8
tests/conftest.py
Normal file
@@ -0,0 +1,8 @@
|
||||
from fastapi.testclient import TestClient
|
||||
import pytest
|
||||
from personal_info.app import create_app
|
||||
|
||||
@pytest.fixture()
|
||||
def app_client():
|
||||
app = create_app()
|
||||
yield TestClient(app)
|
||||
3
tests/test_app.py
Normal file
3
tests/test_app.py
Normal file
@@ -0,0 +1,3 @@
|
||||
def test_health(app_client):
|
||||
rv = app_client.get('/health')
|
||||
assert rv.status_code == 200
|
||||
Reference in New Issue
Block a user