iatoolkit 0.4.2__py3-none-any.whl → 0.66.2__py3-none-any.whl

iatoolkit/views/login_view.py

@@ -0,0 +1,153 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import (request, redirect, render_template, url_for,
+                   render_template_string, flash)
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.jwt_service import JWTService
+from iatoolkit.services.query_service import QueryService
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.i18n_service import I18nService
+from iatoolkit.views.base_login_view import BaseLoginView
+import logging
+
+
+class LoginView(BaseLoginView):
+    """
+    Handles login for local users.
+    Authenticates and then delegates the path decision (fast/slow) to the base class.
+    """
+    def post(self, company_short_name: str):
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        branding_data = self.branding_service.get_company_branding(company)
+        email = request.form.get('email')
+        password = request.form.get('password')
+
+        # 1. Authenticate internal user
+        auth_response = self.auth_service.login_local_user(
+            company_short_name=company_short_name,
+            email=email,
+            password=password
+        )
+
+        if not auth_response['success']:
+            flash(auth_response["message"], 'error')
+            home_template = self.utility.get_company_template(company_short_name, "home.html")
+
+            return render_template_string(
+                home_template,
+                company_short_name=company_short_name,
+                company=company,
+                branding=branding_data,
+                form_data={"email": email},
+            ), 400
+
+        user_identifier = auth_response['user_identifier']
+
+        # 3. define URL to call when slow path is finished
+        target_url = url_for('finalize_no_token',
+                             company_short_name=company_short_name,
+                             _external=True)
+
+        # 2. Delegate the path decision to the centralized logic.
+        try:
+            return self._handle_login_path(company, user_identifier, target_url)
+        except Exception as e:
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500
+
+
+class FinalizeContextView(MethodView):
+    """
+    Finalizes context loading in the slow path.
+    This view is invoked by the iframe inside onboarding_shell.html.
+    """
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 query_service: QueryService,
+                 prompt_service: PromptService,
+                 branding_service: BrandingService,
+                 onboarding_service: OnboardingService,
+                 jwt_service: JWTService,
+                 i18n_service: I18nService
+                 ):
+        self.profile_service = profile_service
+        self.jwt_service = jwt_service
+        self.query_service = query_service
+        self.prompt_service = prompt_service
+        self.branding_service = branding_service
+        self.onboarding_service = onboarding_service
+        self.i18n_service = i18n_service
+
+    def get(self, company_short_name: str, token: str = None):
+        try:
+            session_info = self.profile_service.get_current_session_info()
+            if session_info:
+                # session exists, internal user
+                user_identifier = session_info.get('user_identifier')
+                token = ''
+            elif token:
+                # user identified by api-key
+                payload = self.jwt_service.validate_chat_jwt(token)
+                if not payload:
+                    logging.warning("Fallo crítico: No se pudo leer el auth token.")
+                    return redirect(url_for('home', company_short_name=company_short_name))
+
+                user_identifier = payload.get('user_identifier')
+            else:
+                logging.warning("Fallo crítico: missing session information or auth token")
+                return redirect(url_for('home', company_short_name=company_short_name))
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                            company_short_name=company_short_name,
+                            message="Empresa no encontrada"), 404
+            branding_data = self.branding_service.get_company_branding(company)
+
+            # 2. Finalize the context rebuild (the heavy task).
+            self.query_service.finalize_context_rebuild(
+                company_short_name=company_short_name,
+                user_identifier=user_identifier
+            )
+
+            # 3. render the chat page.
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
+
+            # Get the entire 'js_messages' block in the correct language.
+            js_translations = self.i18n_service.get_translation_block('js_messages')
+
+            return render_template(
+                "chat.html",
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                branding=branding_data,
+                prompts=prompts,
+                onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
+                redeem_token=token
+            )
+
+        except Exception as e:
+            return render_template("error.html",
+                                   company_short_name=company_short_name,
+                                   branding=branding_data,
+                                   message=f"An unexpected error occurred during context loading: {str(e)}"), 500
+

iatoolkit/views/logout_api_view.py

@@ -0,0 +1,49 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import redirect, url_for, jsonify
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.common.session_manager import SessionManager
+import logging
+
+class LogoutApiView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 auth_service: AuthService):
+        self.profile_service = profile_service
+        self.auth_service = auth_service
+
+    def get(self, company_short_name: str = None):
+        try:
+            # 1. Get the authenticated user's
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return jsonify({"error": "company not found."}), 404
+
+            # get URL for redirection
+            url_for_redirect = company.parameters.get('external_urls', {}).get('logout_url')
+            if not url_for_redirect:
+                url_for_redirect = url_for('home', company_short_name=company_short_name)
+
+            # clear de session cookie
+            SessionManager.clear()
+
+            return {
+                'status': 'success',
+                'url': url_for_redirect,
+            }, 200
+        except Exception as e:
+            logging.exception(f"Unexpected error: {e}")
+            return {'status': 'error'}, 500
+
+

iatoolkit/views/profile_api_view.py

@@ -0,0 +1,46 @@
+# iatoolkit/views/profile_api_view.py
+from flask import request, jsonify
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+
+
+class UserLanguageApiView(MethodView):
+    """
+    API endpoint for managing user language preferences.
+    """
+
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService):
+        self.auth_service = auth_service
+        self.profile_service = profile_service
+
+    def post(self):
+        """
+        Handles POST requests to update the user's preferred language.
+        Expects a JSON body with a 'language' key, e.g., {"language": "en"}.
+        """
+        # 1. Authenticate the user from the current session.
+        auth_result = self.auth_service.verify()
+        if not auth_result.get("success"):
+            return jsonify(auth_result), auth_result.get("status_code")
+
+        user_identifier = auth_result.get('user_identifier')
+
+        # 2. Validate request body
+        data = request.get_json()
+        if not data or 'language' not in data:
+            return jsonify({"error_message": "Missing 'language' field in request body"}), 400
+
+        new_lang = data.get('language')
+
+        # 3. Call the service to perform the update
+        update_result = self.profile_service.update_user_language(user_identifier, new_lang)
+
+        if not update_result.get('success'):
+            return jsonify(update_result), 400
+
+        return jsonify({"message": "Language preference updated successfully"}), 200

iatoolkit/views/prompt_api_view.py

@@ -0,0 +1,37 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import jsonify
+from flask.views import MethodView
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+
+class PromptApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 prompt_service: PromptService ):
+        self.auth_service = auth_service
+        self.prompt_service = prompt_service
+
+    def get(self, company_short_name):
+        try:
+            # get access credentials
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get('status_code')
+
+            response = self.prompt_service.get_user_prompts(company_short_name)
+            if "error" in response:
+                return {'error_message': response["error"]}, 402
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"unexpected error getting company prompts: {e}")
+            return jsonify({"error_message": str(e)}), 500

iatoolkit/views/signup_view.py

@@ -0,0 +1,94 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, request, url_for, redirect, flash
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from injector import inject
+from itsdangerous import URLSafeTimedSerializer
+import os
+
+
+class SignupView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service # 3. Guardar la instancia
+        self.i18n_service = i18n_service
+
+        self.serializer = URLSafeTimedSerializer(os.getenv("USER_VERIF_KEY"))
+
+
+    def get(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        branding_data = self.branding_service.get_company_branding(company)
+        return render_template('signup.html',
+                               company=company,
+                               company_short_name=company_short_name,
+                               branding=branding_data)
+
+    def post(self, company_short_name: str):
+        try:
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                                     message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+            branding_data = self.branding_service.get_company_branding(company)
+
+            first_name = request.form.get('first_name')
+            last_name = request.form.get('last_name')
+            email = request.form.get('email')
+            password = request.form.get('password')
+            confirm_password = request.form.get('confirm_password')
+
+            # create verification token and url for verification
+            token = self.serializer.dumps(email, salt='email-confirm')
+            verification_url = url_for('verify_account',
+                                       company_short_name=company_short_name,
+                                       token=token, _external=True)
+
+            response = self.profile_service.signup(
+                company_short_name=company_short_name,
+                email=email,
+                first_name=first_name, last_name=last_name,
+                password=password, confirm_password=confirm_password,
+                verification_url=verification_url)
+
+            if "error" in response:
+                flash(response["error"], 'error')
+                return render_template(
+                    'signup.html',
+                    company=company,
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    form_data={
+                        "first_name": first_name,
+                        "last_name": last_name,
+                        "email": email,
+                        "password": password,
+                        "confirm_password": confirm_password
+                    }), 400
+
+            flash(response["message"], 'success')
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        except Exception as e:
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500

iatoolkit/views/tasks_api_view.py

@@ -0,0 +1,72 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import request, jsonify
+from iatoolkit.services.tasks_service import TaskService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+from datetime import datetime
+import logging
+from typing import Optional
+
+
+class TaskApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 task_service: TaskService,
+                 profile_repo: ProfileRepo):
+        self.auth_service = auth_service
+        self.task_service = task_service
+        self.profile_repo = profile_repo
+
+    def post(self):
+        try:
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            req_data = request.get_json()
+            files = request.files.getlist('files')
+
+            required_fields = ['company', 'task_type', 'client_data']
+            for field in required_fields:
+                if field not in req_data:
+                    return jsonify({"error": f"El campo {field} es requerido"}), 400
+
+            company_short_name = req_data.get('company', '')
+            task_type = req_data.get('task_type', '')
+            client_data = req_data.get('client_data', {})
+            company_task_id = req_data.get('company_task_id', 0)
+            execute_at = req_data.get('execute_at', None)
+
+            # validate date format is parameter is present
+            if execute_at:
+                try:
+                    # date in iso format
+                    execute_at = datetime.fromisoformat(execute_at)
+                except ValueError:
+                    return jsonify({
+                        "error": "El formato de execute_at debe ser YYYY-MM-DD HH:MM:SS"
+                    }), 400
+
+            new_task = self.task_service.create_task(
+                company_short_name=company_short_name,
+                task_type_name=task_type,
+                client_data=client_data,
+                company_task_id=company_task_id,
+                execute_at=execute_at,
+                files=files)
+
+            return jsonify({
+                "task_id": new_task.id,
+                "status": new_task.status.name
+            }), 201
+
+        except Exception as e:
+            logging.exception("Error al crear la tarea: %s", str(e))
+            return jsonify({"error": str(e)}), 500

iatoolkit/views/tasks_review_api_view.py

@@ -0,0 +1,55 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import request, jsonify
+from iatoolkit.services.tasks_service import TaskService
+from iatoolkit.repositories.profile_repo import ProfileRepo
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+from typing import Optional
+
+
+class TaskReviewApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 task_service: TaskService,
+                 profile_repo: ProfileRepo):
+        self.auth_service = auth_service
+        self.task_service = task_service
+        self.profile_repo = profile_repo
+
+    def post(self, task_id: int):
+        auth_result = self.auth_service.verify(anonymous=True)
+        if not auth_result.get("success"):
+            return jsonify(auth_result), auth_result.get("status_code")
+
+        try:
+            req_data = request.get_json()
+            required_fields = ['review_user', 'approved']
+            for field in required_fields:
+                if field not in req_data:
+                    return jsonify({"error": f"El campo {field} es requerido"}), 400
+
+            review_user = req_data.get('review_user', '')
+            approved = req_data.get('approved', False)
+            comment = req_data.get('comment', '')
+
+            new_task = self.task_service.review_task(
+                task_id=task_id,
+                review_user=review_user,
+                approved=approved,
+                comment=comment)
+
+            return jsonify({
+                "task_id": new_task.id,
+                "status": new_task.status.name
+            }), 200
+
+        except Exception as e:
+            logging.exception("Error al revisar la tarea: %s", str(e))
+            return jsonify({"error": str(e)}), 500

iatoolkit/views/user_feedback_api_view.py

@@ -0,0 +1,60 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import request, jsonify
+from flask.views import MethodView
+from iatoolkit.services.user_feedback_service import UserFeedbackService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+
+class UserFeedbackApiView(MethodView):
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 user_feedback_service: UserFeedbackService ):
+        self.auth_service = auth_service
+        self.user_feedback_service = user_feedback_service
+
+    def post(self, company_short_name):
+        try:
+            # get access credentials
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            user_identifier = auth_result.get('user_identifier')
+
+            data = request.get_json()
+            if not data:
+                return jsonify({"error_message": "invalid json body"}), 402
+
+            # these validations are performed also in the frontend
+            # the are localized in the front
+            message = data.get("message")
+            if not message:
+                return jsonify({"error_message": "missing feedback message"}), 400
+
+            rating = data.get("rating")
+            if not rating:
+                return jsonify({"error_message": "missing rating"}), 400
+
+            response = self.user_feedback_service.new_feedback(
+                company_short_name=company_short_name,
+                message=message,
+                user_identifier=user_identifier,
+                rating=rating
+            )
+
+            if "error" in response:
+                return {'error_message': response["error"]}, 402
+
+            return response, 200
+        except Exception as e:
+            logging.exception(
+                f"unexpected error processing feedback for {company_short_name}: {e}")
+            return jsonify({"error_message": str(e)}), 500
+

iatoolkit/views/verify_user_view.py

@@ -0,0 +1,62 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, url_for, redirect, session, flash
+from iatoolkit.services.profile_service import ProfileService
+from itsdangerous import URLSafeTimedSerializer, SignatureExpired
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from injector import inject
+import os
+
+
+class VerifyAccountView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.i18n_service = i18n_service
+        self.serializer = URLSafeTimedSerializer(os.getenv("USER_VERIF_KEY"))
+
+    def get(self, company_short_name: str, token: str):
+        try:
+            # get company info
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                                       message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+            branding_data = self.branding_service.get_company_branding(company)
+            try:
+                # decode the token from the URL
+                email = self.serializer.loads(token, salt='email-confirm', max_age=3600*5)
+            except SignatureExpired:
+                flash(self.i18n_service.t('errors.verification.token_expired'), 'error')
+                return render_template('signup.html',
+                                       company=company,
+                                       company_short_name=company_short_name,
+                                       branding=branding_data,
+                                       token=token), 400
+
+            response = self.profile_service.verify_account(email)
+            if "error" in response:
+                flash(response["error"], 'error')
+                return render_template(
+                    'signup.html',
+                    company=company,
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    token=token), 400
+
+            flash(response['message'], 'success')
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        except Exception as e:
+            flash(self.i18n_service.t('errors.general.unexpected_error'), 'error')
+            return redirect(url_for('home', company_short_name=company_short_name))

{iatoolkit-0.4.2.dist-info → iatoolkit-0.66.2.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.4.2
+Version: 0.66.2
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT
-Requires-Python: >=3.11
+Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 Requires-Dist: aiohappyeyeballs==2.4.4
 Requires-Dist: aiohttp==3.11.9