diff --git a/app/api/services/office_templates.py b/app/api/services/office_templates.py index 1ec3f30..469d720 100644 --- a/app/api/services/office_templates.py +++ b/app/api/services/office_templates.py @@ -4,7 +4,7 @@ from app.api.services.base_service import BaseService from sqlalchemy import func, desc from datetime import datetime, timedelta, timezone from app.extensions import db -from app.models import OfficeUsers, CustomTemplates +from app.models import OfficeUsers, CustomTemplates, MembersProducts, ProductsTemplates class OfficeTemplatesService(BaseService): @@ -45,20 +45,21 @@ class OfficeTemplatesService(BaseService): def set_office_user_template(data): logger.info('set_office_user_template') try: - # office_users = OfficeUsers.get_office_users_list() - # office_users_data = [] - # if office_users: - # for t in office_users: - # office_users_data.append({ - # 'id': t.id, - # 'uid': t.uid, - # 'username': t.username, - # 'firstname': t.firstname, - # 'lastname': t.lastname, - # 'acc_level': t.acc_level, - # 'status': t.status, - # 'added': t.added.isoformat() if t.added else None - # }) + subscrtiption_uid = data.get('subscrtiption_uid', '') + template_uid = data.get('template_uid', '') + selectProduct = None + selectedTemplate = None + + selectProduct = MembersProducts.get_member_product_by_subscription_uid(subscrtiption_uid) + if selectProduct: + selectedTemplate = ProductsTemplates.get_template_by_uid(template_uid) + + if selectedTemplate is None or selectedTemplate is None: + return jsonify({ + 'error': 'Missing credentials', + 'message': 'Invalid credentials', + }), 400 + office_users_result = { "set_office_user_template": [], diff --git a/app/models/products_templates.py b/app/models/products_templates.py index bd9db4f..f61d5bd 100644 --- a/app/models/products_templates.py +++ b/app/models/products_templates.py @@ -34,6 +34,13 @@ class ProductsTemplates(db.Model): raise ValueError(f"Templates not found") return templates + @classmethod + def get_template_by_uid(cls, template_uid): + selTemplate = cls.query.filter_by(uif=template_uid).first() + if not selTemplate: + return None + return selTemplate + def to_dict(self): return {