fixed incorrect typing issues

This commit is contained in:
2023-11-26 13:41:42 +01:00
parent 530b397592
commit 80ccaa83d3
8 changed files with 92 additions and 60 deletions

View File

@@ -1,13 +1,20 @@
from typing import List
from sqlalchemy.orm import Session
from . import sql_models, sql_schemas
from . import api_outputs, sql_models
def get_producers(db: Session, skip: int = 0, limit: int = 100):
def get_producers(
db: Session, skip: int = 0, limit: int = 100
) -> List[sql_models.Producer]:
return db.query(sql_models.Producer).offset(skip).limit(limit).all()
def create_producer(db: Session, producer: sql_schemas.ProducerCreate):
jsonblob_init = {"test_repo":"jsonblob_create"}
def create_producer(
db: Session, producer: api_outputs.ProducerCreate
) -> sql_models.Producer:
jsonblob_init = {"test_repo": "jsonblob_create"}
db_producer = sql_models.Producer(jsonblob=jsonblob_init)
db.add(db_producer)
db.commit()
@@ -15,12 +22,17 @@ def create_producer(db: Session, producer: sql_schemas.ProducerCreate):
return db_producer
def get_repositories(db: Session, skip: int = 0, limit: int = 100):
def get_repositories(
db: Session, skip: int = 0, limit: int = 100
) -> List[sql_models.Repository]:
return db.query(sql_models.Repository).offset(skip).limit(limit).all()
def create_repository(db: Session, repository: sql_schemas.RepositoryCreate, producers_id: int):
def create_repository(
db: Session, repository: api_outputs.RepositoryCreate, producers_id: int
) -> sql_models.Repository:
db_repository = sql_models.Repository(**repository.dict(), prod_id=producers_id)
db.add(db_repository)
db.commit()
db.refresh(db_repository)
return db_repository
return db_repository