codekingpro/portable-devtools
114k
1�
2 �j� � �� � S r SSKrSSKrSSKJr SSKJrJrJrJ r SSK J r
" S S\R 5 rS rS3 r\S\
4S j5 r\S 5 rg! \4 a N>f = f)
zB5Helper functionality for interoperability with stdlib `logging`.6� N)�contextmanager)�Iterator�List�Optional�Type� )�tqdmc �4 ^ � \ rS rSr\4U 4S jjrS rSrU =r$ )�_TqdmLoggingHandler� c �. >� [ TU ] 5 Xl g �N)�super�__init__�7tqdm_class)�selfr � __class__s ��ZD:\code\apps\devtools\python\user_packages\Python313\site-packages\tqdm/contrib/logging.pyr � _TqdmLoggingHandler.__init__ s �� � ����$�� c �� � U R U5 nU R R X R S9 U R 5 g ! [8 [ 4 a e U R U5 g = f)N)�file)�formatr �write�stream�flush�KeyboardInterrupt�9SystemExit� handleError)r �record�msgs r �emit�_TqdmLoggingHandler.emit s\ � � %��+�+�f�%�C� �O�O� !� !�#�K�K� !� 8� �J�J�L��!�:�.� � � %� � � �V� $�s �AA �#A,)r ) �__name__�10__module__� __qualname__�__firstlineno__�std_tqdmr r" �__static_attributes__�
__classcell__)r s @r r r s �� � �%�%� %r r c � � [ U [ R 5 =( a- U R [ R11 [ R 1; $ r )�12isinstance�logging�
StreamHandlerr �sys�stdout�stderr)�handlers r �_is_console_logging_handlerr3 # s5 � � �w�� 5� 5� 6�
;����3�:�:�s�z�z�":�:�<r c �> � U H n[ U5 ( d M Us $ g r )r3 )�handlersr2 s r �(_get_first_found_console_logging_handlerr6 ( s � ��� &�w� /� /��N� r c # � # � U c [ R /n U Vs/ s H o"R PM nn U H� n[ U5 n[ UR 5 nUb, UR UR 5 UR Ul UR Vs/ s H n[ U5 ( a M UPM snU/- Ul M� Sv � [ X5 H u p'Xrl M
gs snf s snf ! [ X5 H u p'Xrl M
f = f7f)a� 13Context manager redirecting console logging to `tqdm.write()`, leaving14other logging handlers (e.g. log files) unaffected.15 16Parameters17----------18loggers : list, optional19 Which handlers to redirect (default: [logging.root]).20tqdm_class : optional21 22Example23-------24```python25import logging26from tqdm import trange27from tqdm.contrib.logging import logging_redirect_tqdm28 29LOG = logging.getLogger(__name__)30 31if __name__ == '__main__':32 logging.basicConfig(level=logging.INFO)33 with logging_redirect_tqdm():34 for i in trange(9):35 if i == 4:36 LOG.info("console logging redirected to `tqdm.write()`")37 # logging restored38```39N)40r- �rootr5 r r6 � setFormatter� formatterr r3 �zip)�loggersr �logger�original_handlers_list� tqdm_handler� orig_handlerr2 �original_handlerss r �logging_redirect_tqdmrB . s� � � �D ���<�<�.��<C� D�G�&�o�o�G�� D�
0� �F�.�z�:�L�C�F�O�O�T�L��'� �)�)�,�*@�*@�A�&2�&9�&9� �#�'-��� =�'6�G�2�7�;� �� =�@L�~� N�F�O�
� �),�W�)M� %�F� /�O� *N�� E�� =��41 *-�W�)M� %�F� /�O� *N�s? �D �C"�D �A"C, �C'�-C'�3C, � D �'C, �, D�D c / � # � UR 5 nUR SS5 nUR S[ 5 nU" U 0 UD6 n[ X4S9 Uv � SSS5 SSS5 g! , ( d f N= f! , ( d f g= f7f)a@ 42Convenience shortcut for:43```python44with tqdm_class(*args, **tqdm_kwargs) as pbar:45 with logging_redirect_tqdm(loggers=loggers, tqdm_class=tqdm_class):46 yield pbar47```48 49Parameters50----------51tqdm_class : optional, (default: tqdm.std.tqdm).52loggers : optional, list.53**tqdm_kwargs : passed to `tqdm_class`.54r< Nr )r< r )�copy�popr( rB )�args�kwargs� tqdm_kwargsr< r �pbars r �tqdm_logging_redirectrJ c sk � � �, �+�+�-�K��o�o�i��.�G� � � � �x�8�J� �T� )�[� )�T�
"�7�
J��J� K� 55*� )�
J�
J�� 56*� )�s6 �AB�57A4�
A#�A4� B�#58A1 �-A4�459B�>B)�__doc__r- r/ �60contextlibr �typingr r r r � ImportError�stdr r( r. r r3 r6 rB rJ � r r �<module>rQ s� ��� � 61� %� �5�5� #�%�'�/�/� %�&<�62� � ��10� �10�h �� ���u � �� �s � A �A � A