acazares
52b8fcd434
- Enhanced TenantMiddleware to validate tenant information from JWT tokens. - Added LicenseValidationMiddleware to check tenant licenses before processing requests. - Updated security utilities to extract tenant information from tokens and validate company access. - Introduced CompanyStore to manage active company state and handle company switching in the frontend. - Modified API routes to include company_id in requests for better resource management. - Improved logging and error handling throughout the middleware and API layers. - Updated frontend components to reflect changes in company management and selection. - Added new API route for fetching user's companies with proper authentication handling.
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
|
Service layer for CountryRuleOct.
|
|
"""
|
|
|
|
from sqlalchemy.orm import Session
|
|
from . import models, dto
|
|
|
|
|
|
class CountryRuleOctService:
|
|
@staticmethod
|
|
def get_country_by_keys(
|
|
db: Session, permission: str, line: int, fraction: str, country_code: str
|
|
):
|
|
return (
|
|
db.query(models.CountryRuleOct)
|
|
.filter(
|
|
models.CountryRuleOct.permission == permission,
|
|
models.CountryRuleOct.line == line,
|
|
models.CountryRuleOct.fraction == fraction,
|
|
models.CountryRuleOct.country_code == country_code,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create_country_rule(db: Session, country_data: dto.CountryRuleOctCreateDTO):
|
|
new_country = models.CountryRuleOct(**country_data.dict())
|
|
db.add(new_country)
|
|
db.commit()
|
|
db.refresh(new_country)
|
|
return new_country
|
|
|
|
@staticmethod
|
|
def delete_country_rule(
|
|
db: Session, permission: str, line: int, fraction: str, country_code: str
|
|
):
|
|
country = CountryRuleOctService.get_country_by_keys(
|
|
db, permission, line, fraction, country_code
|
|
)
|
|
if country:
|
|
db.delete(country)
|
|
db.commit()
|
|
return country
|