cmdbox 0.5.3.1__py3-none-any.whl → 0.6.0__py3-none-any.whl

cmdbox/app/auth/signin_saml.py

@@ -0,0 +1,61 @@
+from cmdbox.app.auth.signin import Signin
+from fastapi import Request, Response
+from typing import Any, Dict, Tuple
+import copy
+import logging
+
+
+class SigninSAML(Signin):
+
+    def jadge(self, email:str) -> Tuple[bool, Dict[str, Any]]:
+        """
+        サインインを成功させるかどうかを判定します。
+        返すユーザーデータには、uid, name, email, groups, hash が必要です。
+
+        Args:
+            email (str): メールアドレス
+
+        Returns:
+            Tuple[bool, Dict[str, Any]]: (成功かどうか, ユーザーデータ)
+        """
+        copy_signin_data = copy.deepcopy(self.signin_file_data)
+        users = [u for u in copy_signin_data['users'] if u['email'] == email and u['hash'] == 'saml']
+        return len(users) > 0, users[0] if len(users) > 0 else None
+
+    async def make_saml(self, prov:str, next:str, form_data:Dict[str, Any], req:Request, res:Response) -> Any:
+        """
+        SAML認証のリダイレクトURLを取得する
+        Args:
+            prov (str): プロバイダ名
+            next (str): リダイレクト先のURL
+            req (Request): リクエスト
+            res (Response): レスポンス
+        Returns:
+            OneLogin_Saml2_Auth: SAML認証オブジェクト
+        """
+        sd = self.get_data()
+        saml_settings = dict(
+            strict=False,
+            debug=self.logger.level==logging.DEBUG,
+            idp=sd['saml']['providers'][prov]['idp'],
+            sp=sd['saml']['providers'][prov]['sp'])
+        # SAML認証のリダイレクトURLを取得
+        request_data = dict(
+            https='on' if req.url.scheme=='https' else 'off',
+            http_host=req.client.host,
+            server_port=req.url.port,
+            script_name=f'{req.url.path}?next={next}',
+            post_data=dict(),
+            get_data=dict(),
+        )
+        if (req.query_params):
+            request_data["get_data"] = req.query_params,
+        if "SAMLResponse" in form_data:
+            SAMLResponse = form_data["SAMLResponse"]
+            request_data["post_data"]["SAMLResponse"] = SAMLResponse
+        if "RelayState" in form_data:
+            RelayState = form_data["RelayState"]
+            request_data["post_data"]["RelayState"] = RelayState
+        from onelogin.saml2.auth import OneLogin_Saml2_Auth
+        auth = OneLogin_Saml2_Auth(request_data=request_data, old_settings=saml_settings)
+        return auth

cmdbox/app/common.py

@@ -4,6 +4,8 @@ from cmdbox.app.commons import convert, module, loghandler
 from cryptography.fernet import Fernet
 from pathlib import Path
 from pkg_resources import resource_string
+from rich.logging import RichHandler
+from rich.console import Console
 from tabulate import tabulate
 from typing import List, Tuple, Dict, Any
 import asyncio
@@ -91,6 +93,48 @@ def save_yml(yml_path:Path, data:dict) -> None:
     with open(yml_path, 'w') as f:
         yaml.dump(data, f, default_flow_style=False, sort_keys=False)
 
+def reset_logger(name:str, stderr:bool=False, fmt:str='%(message)s') -> None:
+    """
+    指定されたロガーのハンドラをクリアし、新しいハンドラを追加します。
+    Args:
+        name (str): ロガーの名前
+        stderr (bool, optional): 標準エラー出力を使用するかどうか. Defaults to False.
+        fmt (str, optional): ログフォーマット. Defaults to '%(message)s'.
+    """
+    logger = logging.getLogger(name)
+    logger.handlers.clear()
+    logger.propagate = False
+    logger.addHandler(create_log_handler(stderr, fmt))
+
+def create_log_handler(stderr:bool=False, fmt:str='%(message)s') -> logging.Handler:
+    """
+    ログハンドラを生成します。
+
+    Args:
+        stderr (bool, optional): 標準エラー出力を使用するかどうか. Defaults to False.
+        fmt (str, optional): ログフォーマット. Defaults to '%(message)s'.
+    Returns:
+        logging.Handler: ログハンドラ
+    """
+    formatter = logging.Formatter(fmt)
+    handler = RichHandler(console=Console(stderr=stderr), show_path=False, omit_repeated_times=False,
+                          tracebacks_word_wrap=False, log_time_format='[%Y-%m-%d %H:%M]')
+    handler.setFormatter(formatter)
+    return handler
+
+def create_console(stderr:bool=False, file=None) -> Console:
+    """
+    コンソールを生成します。
+
+    Args:
+        stderr (bool, optional): 標準エラー出力を使用するかどうか. Defaults to False.
+        file (file, optional): 出力先のファイル. Defaults to None.
+    Returns:
+        Console: コンソール
+    """
+    console = Console(stderr=stderr, file=file, log_time=True, log_path=False, log_time_format='[%Y-%m-%d %H:%M]')
+    return console
+
 def default_logger(debug:bool=False, ver=version, webcall:bool=False) -> logging.Logger:
     """
     デフォルトのロガーを生成します。
@@ -105,12 +149,13 @@ def default_logger(debug:bool=False, ver=version, webcall:bool=False) -> logging
     """
     logger = logging.getLogger(ver.__appid__)
     if not webcall:
-        formatter = logging.Formatter('%(levelname)s[%(asctime)s] - %(message)s')
-        handler = loghandler.ColorfulStreamHandler(sys.stdout)
+        #formatter = logging.Formatter('%(levelname)s[%(asctime)s] - %(message)s')
+        #handler = loghandler.ColorfulStreamHandler(sys.stdout)
+        handler = create_log_handler()
         handler.setLevel(logging.DEBUG if debug else logging.INFO)
-        handler.setFormatter(formatter)
         logger.addHandler(handler)
     logger.setLevel(logging.DEBUG if debug else logging.INFO)
+    logger.propagate = False
     return logger
 
 def load_config(mode:str, debug:bool=False, data=HOME_DIR, webcall:bool=False, ver=version) -> Tuple[logging.Logger, dict]: