ceph/src/pybind/mgr/dashboard/services/custom_banner.py

   1 import logging
   2 from typing import Optional
   3
   4 from mgr_module import _get_localized_key
   5
   6 from .. import mgr
   7
   8 logger = logging.getLogger(__name__)
   9
  10
  11 def set_login_banner_mgr(inbuf: str, mgr_id: Optional[str] = None):
  12     item_key = 'custom_login_banner'
  13     if mgr_id is not None:
  14         mgr.set_store(_get_localized_key(mgr_id, item_key), inbuf)
  15     else:
  16         mgr.set_store(item_key, inbuf)
  17
  18
  19 def get_login_banner_mgr():
  20     banner_text = mgr.get_store('custom_login_banner')
  21     logger.info('Reading custom login banner: %s', banner_text)
  22     return banner_text
  23
  24
  25 def unset_login_banner_mgr():
  26     mgr.set_store('custom_login_banner', None)
  27     logger.info('Removing custom login banner')