from sqlalchemy.orm import Session
from app.db import models


def get_market_by_name(db: Session, name: str):
    return db.query(models.Market).filter(models.Market.name == name).first()

def create_market(db: Session, name: str):
    db_obj = models.Market(name=name)
    db.add(db_obj)
    db.commit()
    db.refresh(db_obj)
    return db_obj

def get_sector_by_name(db: Session, name: str):
    return db.query(models.Sector).filter(models.Sector.name == name).first()

def create_sector(db: Session, name: str):
    db_obj = models.Sector(name=name)
    db.add(db_obj)
    db.commit()
    db.refresh(db_obj)
    return db_obj