cache.cpython-313.pyc•7.57 kB
�
��7h� � �\ � S r SSKrSSKrSSKJrJrJr SSKJrJr SSK J
r
" S S5 r g)z-Caching utilities for URL reputation checker.� N)�Optional�Any�Dict)�datetime� timedelta)�Redisc �� � \ rS rSrSrSS\4S jjrS rS rS r S\S \S
\4S jr
S \S
\ \ \\
4 4S
jrS \S\ \\
4 S\4S jrS\S
\ \ \\
4 4S jrS\S\ \\
4 4S jrS
\ \\4 4S jrS rSrg)� CacheManager� z*Manage caching for URL validation results.� redis_urlc �H � Xl S U l SU l SU l SU l g )Ni�Q i i�: )r �redis� ttl_valid� ttl_invalid� ttl_history)�selfr s �S/home/josh/Projects/reputation-and-validity-checker/url_reputation_checker/cache.py�__init__�CacheManager.__init__ s% � �"��&*��
� ��� ���!��� c � �� # � [ R " U R 5 U l U R R 5 I Sh v�N g N! [
a
SU l gf = f7f)zConnect to Redis.N)r �from_urlr r �ping� Exception�r s r �connect�CacheManager.connect sG � � � � � � ����7�D�J��*�*�/�/�#� #� #��� � �D�J� �s; �A%�AA �A �A � A%� A �A"� A%�!A"�"A%c � �\ # � U R c U R 5 I Sh v�N gg N7f)z Ensure we're connected to Redis.N)r r r s r �_ensure_connected� CacheManager._ensure_connected s& � � � �:�:� ��,�,�.� � � � �s �!,�*�,c � �x # � U R ( a# U R R 5 I Sh v�N gg N7f)zDisconnect from Redis.N)r �closer s r �
disconnect�CacheManager.disconnect# s* � � � �:�:��*�*�"�"�$� $� $� � $�s �/:�8�:�prefix�
identifier�returnc � � [ U5 S:� a2 [ R " UR 5 5 R 5 nSU SU 3$ )zGenerate cache key.�� zurl_reputation:�:)�len�hashlib�md5�encode� hexdigest)r r% r& s r �_get_cache_key�CacheManager._get_cache_key( sC � � �z�?�S� � ���Z�%6�%6�%8�9�C�C�E�J� ����*� �6�6r �urlc � �� # � U R ( d g U R SU5 nU R R U5 I Sh v�N nU( a [ R " U5 $ g N#! [
a gf = f7f)z Get cached validation result.N�
validation�r r0 �get�json�loadsr )r r2 �key�datas r �get_validation_result�"CacheManager.get_validation_result/ sr � � ��z�z�� ��%�%�l�C�8�C� ��� � ��,�,�D���z�z�$�'�'� �
�
-�� � � �� ��: �A<�0A, �A*� A, �'A<�*A, �,
A9�6A<�8A9�9A<�result�is_validc � �v # � U R ( d g U R SU5 nU( a U R O U R n[ R
" 5 R
5 US' U R R UU[ R " U5 5 I Sh v�N g N! [ a gf = f7f)zCache validation result.Nr4 � cached_at) r r0 r r r �utcnow� isoformat�setexr7 �dumpsr )r r2 r> r? r9 �ttls r �set_validation_result�"CacheManager.set_validation_result> s� � � ��z�z� �
��%�%�l�C�8�C�$,�$�.�.�$�2B�2B�C� #+�/�/�"3�"=�"=�"?�F�;� ��*�*�"�"����
�
�6�"��
�
��
� � � �s; �B9�B B) �!B'�"B) �&B9�'B) �)
B6�3B9�5B6�6B9�domainc � �� # � U R ( d g U R SU5 nU R R U5 I Sh v�N nU( a [ R " U5 $ g N#! [
a gf = f7f)zGet cached domain history.N�historyr5 )r rI r9 r: s r �get_domain_history� CacheManager.get_domain_historyR sr � � ��z�z�� ��%�%�i��8�C� ��� � ��,�,�D���z�z�$�'�'� �
�
-�� � � �� �r= rK c � �L # � U R ( d g U R SU5 n[ R " 5 R 5 US' U R R UU R [ R " U5 5 I Sh v�N g N! [ a gf = f7f)zCache domain history.NrK rA )
r r0 r rB rC rD r r7 rE r )r rI rK r9 s r �set_domain_history� CacheManager.set_domain_historya s� � � ��z�z� � ��%�%�i��8�C� $,�?�?�#4�#>�#>�#@�G�K� ��*�*�"�"��� � ��
�
�7�#��
�
��
� � � �s; �B$�A6B � B�
B �B$�B �
B!� B$� B!�!B$c � � # � U R 5 I Sh v�N U R ( d SS0$ U R R S5 I Sh v�N nU R R S5 I Sh v�N nS[ U5 [ U5 [ U5 [ U5 - S.$ N� NW N6! [ a SS0s $ f = f7f)zGet cache statistics.N�enabledFzurl_reputation:validation:*zurl_reputation:history:*T)rR �validation_entries�history_entries�
total_entries)r r �keysr+ r )r �validation_keys� history_keyss r � get_stats�CacheManager.get_statst s� � � ��$�$�&�&�&��z�z� �u�%� %� &�$(�J�J�O�O�4Q�$R� R�O�!%��� � �1K�!L�L�L� �&)�/�&:�#&�|�#4�!$�_�!5��L�8I�!I� �
� '�
S�L�� � &� �u�%� %� &�sV �B?�B%�B?� B+ �B'�"B+ �2B)�31B+ �$B?�'B+ �)B+ �+B<�9B?�;B<�<B?c � �& # � U R 5 I Sh v�N U R ( d g U R R S5 I Sh v�N nU( a" U R R " U6 I Sh v�N gg Nd N0 N
! [ a gf = f7f)zClear all cache entries.Nzurl_reputation:*)r r rV �deleter )r rV s r � clear_cache�CacheManager.clear_cache� sz � � ��$�$�&�&�&��z�z� � � ��� � �);�<�<�D���j�j�'�'��.�.�.� �
'�
=�.��� � � �sV �B�A;�B� B � A=�
'B �4A?�5B �9B�=B �?B �
B� B�
B�B)r r r r r N)zredis://localhost:6379)�__name__�
__module__� __qualname__�__firstlineno__�__doc__�strr r r r# r0 r r r r; �boolrG rL rO �intrY r] �__static_attributes__� r r r
r
s� � �4�"�#� "� �!�
%�
7�S� 7�c� 7�c� 7�
�s�
�x��S�#�X��7O�
� �s� �D��c��N� �VZ� �(
�s�
�x��S�#�X��7O�
� �s� �T�#�s�(�^� �&&��c�3�h� � &�& r r
) rc r7 r, �typingr r r r r �
redis.asyncior r
rh r r �<module>rk s& �� 3� � � &� &� (� �G� Gr