EdjSignUtils.cpython-311.pyc•5.22 kB
�
�Qrh� � � � d dl Z d dlmZmZ d dlZd dlmZ d dlmZ d dl Z G d� d� � Z
d� Z e dk r
e � � dgZ
dS ) � N)�Dict�List)�AES)�unpadc �� � e Zd ZdZdZedeeef dedefd�� � Zededefd�� � Z edeeef de
fd �� � Z ede
defd
�� � Z ed edefd �� � Z
d
S )� EdjSignUtilsu
e代驾签名工具类
�1234567890abcdef�params�secret�returnc �� � t � | � � }t � |� � }||z |z }t � |� � dd� }|S )u�
生成签名的方法
:param params: 本次请求的所有参数都放在此dict中
:param secret: e代驾分配的SECRET
:return: 签名字符串
N� )r �sort�create_query_str�md5)r
r � params_list� query_str�md_str�sigs �:/Users/dongfengsu/Dev/edaijiamcp/edjserver/EdjSignUtils.py� generate_sigzEdjSignUtils.generate_sig sZ � � #�'�'��/�/� � �1�1�+�>�>� ��)�#�f�,��� � �v�&�&�s��s�+���
� �
encrypt_tokenc �& � t j t j � � � t j � � }t
j | � � }|� |� � }t |t j
� � � � � }|S )u�
AES解密token
:param encrypt_token: 加密的token字符串
:param random_key: 解密密钥(randomkey参数值)
:return: 解密后的token
) r �newr �DEFAULT_RANDOMKEY�encode�MODE_ECB�base64� b64decode�decryptr �
block_size�decode)r �cipher�encrypted_data�decrypted_data�tokens r �
decrypt_tokenzEdjSignUtils.decrypt_token si � � �� �7�>�>�@�@�#�,�O�O�� �)�-�8�8�� � � ��7�7���n�c�n�5�5�<�<�>�>��� r c �J � t | � � � d� �� � S )u-
对参数字典按key排序
c � � | d S )Nr � )�xs r �<lambda>z#EdjSignUtils.sort.<locals>.<lambda>3 s
� �A�a�D� r )�key)�sorted�items)r
s r r zEdjSignUtils.sort. s! � �
�f�l�l�n�n�.�.�9�9�9�9r c �� � g }| D ]U\ }}|dvrL|� t |� � � � |�(|dk r"|� t |� � � � �Vd� |� � S )u'
创建查询字符串
)� gpsstring�callback�_r N� )�append�str�join)r
�sbr. �values r r z EdjSignUtils.create_query_str5 sw � �
�� � *� *�J�C���?�?�?�� � �#�c�(�(�#�#�#��$��"����I�I�c�%�j�j�)�)�)���w�w�r�{�{�r �
plain_textc � � t j � � }|� | � � � � � |� � � S )uZ
MD5加密
:param plain_text: 明文
:return: 32位密文
)�hashlibr �updater � hexdigest)r; �md5_hashs r r zEdjSignUtils.md5B s? � � �;�=�=�����
�)�)�+�+�,�,�,��!�!�#�#�#r N)�__name__�
__module__� __qualname__�__doc__r � staticmethodr r7 r r( r r r r r+ r r r r s! � � � � � �� � +��� �T�#�s�(�^� �S� �S� � � � �\� � ��S� �S� � � � �\��" �:�T�#�s�(�^� :�� :� :� :� �\�:� �
��
�#�
�
�
� �\�
� �$�� $�� $� $� $� �\�$� $� $r r c �� � dddddddd�} d }t � | |� � }d
}||k sJ d |� d |� �� � �t d
|� �� � t d� � dS )u"
测试签名生成方法
�61000158z2019-06-15 11:57:11�01012345z3.4.3z
116.476169z 40.018682u 北京)�appkey� timestamp�from�ver� longitude�latitude� city_namez$0031186e-5cc6-45a6-a090-3e88ec220452� 7bebf0fe6453861c9d304a83bc0eedu 签名验证失败: 期望值=u , 实际值=u 生成的签名值: u 签名验证通过!N)r r �print)r
r r � expected_sigs r �test_generate_sigrS M s� � � �*���!� � �� �F� 4�F� �
#�
#�F�F�
3�
3�C� 4�L� �,� � � � `� � `� `�[^� `� `� � � � �
&��
&�
&�'�'�'� �
� � � � � r �__main__)r= �typingr r � collections�
Crypto.Cipherr �Crypto.Util.Paddingr r r rS rA �__all__r+ r r �<module>rZ s� �� ���� � � � � � � � � � � � � � � � � � � %� %� %� %� %� %�
�
�
�
�C$� C$� C$� C$� C$� C$� C$� C$�J!� !� !�: �z�������� �
���r