Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file modified db.sqlite
Binary file not shown.
10 changes: 7 additions & 3 deletions library_backend/database.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
import random
import sqlalchemy
import uuid


from datetime import date
from functools import wraps
from faker import Faker
Expand Down Expand Up @@ -39,16 +42,17 @@ def __init__(self):
self.engine = create_engine("sqlite:///db.sqlite", echo=False)
self.session = None
self.connection_name = None
self.inspect = sqlalchemy.inspect(self.engine)

def __enter__(self):
self.session = sessionmaker(bind=self.engine)()

@check_session()
def create_tables_if_not_exists(self):
try:
if not (self.engine.dialect.has_table(self.engine, UsersDBModel.__tablename__)
and self.engine.dialect.has_table(self.engine, BooksDBModel.__tablename__)
and self.engine.dialect.has_table(self.engine, ReservationsDBModel.__tablename__)):
if not (self.inspect.has_table(UsersDBModel.__tablename__, schema=None)
and self.inspect.has_table(BooksDBModel.__tablename__, schema=None)
and self.inspect.has_table(ReservationsDBModel.__tablename__, schema=None)):
logger.info(f"Creating table {UsersDBModel.__tablename__}...")
try:
Base.metadata.create_all(self.engine)
Expand Down
9 changes: 5 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
flask
sqlalchemy
faker
pytest
mock==4.0.3
flask==2.3.2
sqlalchemy==1.4.35
faker==13.3.4
pytest==7.11.1
3 changes: 2 additions & 1 deletion tests/books_endpoints/test_book_service.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
import copy

import mock
import pytest

from unittest import mock

from library_backend.exceptions import BookAlreadyExists, ResourceNotFound
from library_backend.service import BookService, BooksDBModel

Expand Down