1.1.0 添加授权时间支持

Author: lyy289065406

README.md

@@ -65,15 +65,16 @@ u_machine_code = gen_machine_code(my_crypt)
 ### 场景步骤三（管理员本地）
 
 1. 管理员解密用户提供的【加密机器码】文件
-2. 同时为用户分配【随机用户码】文件
+2. 同时为用户设置授权天数，生成【用户码】文件
 3. 结合两者生成【注册码】文件
-4. 把【随机用户码】文件和【注册码】文件提供给用户
+4. 把【用户码】文件和【注册码】文件提供给用户
 
 ```python
 from mur.admin import *
 
 a_machine_code = read_machine_code()
-a_user_code = gen_user_code()
+days = input('请输入授权天数：')    # 0 表示永久
+a_user_code = gen_user_code(days, crypt)
 a_register_code = gen_register_code(
     a_machine_code, a_user_code, my_crypt
 )
@@ -89,7 +90,7 @@ a_register_code = gen_register_code(
 3. 主程序在用户本地重新生成【机器码】
 4. 主程序利用【用户码】和【机器码】生成【注册码】
 5. 主程序比对【生成的注册码】和【管理员提供的注册码】内容是否一致
-6. 若一致，程序运行；否则，程序终止
+6. 若一致，且授权未过期，程序运行；否则，程序终止
 
 ```python
 from mur.user import *

gen_machine_code.py

@@ -19,6 +19,10 @@ def main() :
     print('已生成机器码 [%s]' % machine_code)
     print('已生成机器码对应的本地文件 [%s]' % MACHINE_CODE_PATH)
     print('请将其发送给管理员，以获取注册码。')
+    print('')
+    print('')
+    input('输入回车退出 ...')
+
 
 
 if __name__ == '__main__':

gen_register_code.py

@@ -23,7 +23,8 @@ def main() :
         des_iv = read_des_iv()
         crypt = Crypt(des_key, des_iv)
 
-        user_code = gen_user_code()
+        days = input('请输入授权天数：')
+        user_code = gen_user_code(days, crypt)
         print('随机分配的【用户码】为 [%s]' % user_code)
         print('已把【用户码】保存到根目录的文件 [%s]，请把该文件发送给用户' % USER_CODE_PATH)
 

setup.py

@@ -37,7 +37,7 @@
     # For a discussion on single-sourcing the version across setup.py and the
     # project code, see
     # https://packaging.python.org/en/latest/single_source_version.html
-    version='1.0.6',  # Required. eg. 1.2.3
+    version='1.1.0',  # Required. eg. 1.2.3
 
     # This is a one-line description or tagline of what your project does. This
     # corresponds to the "Summary" metadata field:
@@ -143,7 +143,7 @@
     #
     # For an analysis of "install_requires" vs pip's requirements files see:
     # https://packaging.python.org/en/latest/requirements.html
-    install_requires=['pyDes>=2.0.1', 'psutil>=5.8.0', 'pypiwin32'],  # Optional
+    install_requires=['pyDes>=2.0.1', 'psutil>=5.8.0', 'pypiwin32', 'requests>=2.22.0'],  # Optional
 
     # List additional groups of dependencies here (e.g. development
     # dependencies). Users will be able to install these using the "extras"

src/mur/_public.py

@@ -4,13 +4,19 @@
 
 import os
 import time
+import requests
+import json
+from requests.api import put
 from .crypt import Crypt
 from ._mi import MachineInfo
 
 CRYPT = Crypt()     
 MI = MachineInfo()
 
 JOINER = '##'
+ONLINE_TIME_URL = 'http://api.m.taobao.com/rest/api3.do?api=mtop.common.getTimestamp'
+UNLIMIT = '0000000000'
+
 MACHINE_CODE_PATH = '.machine'
 USER_CODE_PATH = '.user'
 REGISTER_CODE_PATH = '.register'
@@ -36,16 +42,34 @@ def gen_rc(crypt, uuid, user_code) :
     )
 
 
+def now() :
+    '''
+    获取当前时间点（精确到秒）
+    [return] 当前时间戳（long）
+    '''
+    try :
+        # 先获取在线时间，避免用户修改本地时间绕过注册校验
+        response = requests.get(ONLINE_TIME_URL, verify=False, timeout=10)
+        millis = int(json.loads(response.text)['data']['t'])
+        seconds = int(millis / 1000)
+    except :
+        # 网络不通时用本地时间代替
+        seconds = int(time.time())
+    return seconds
+
+
 def after(days) :
     '''
     获取今天之后的 n 天的时间点（精确到秒）
     [param] days: 之后的 n 天
-    [return] n 天后的 long 时间戳
+    [return] n 天后的时间戳（str）
     '''
-    days = 0 if days <= 0 else days
-    seconds = days * 86400
-    now = time.time()
-    return now + seconds
+    timestamp = UNLIMIT
+    if days > 0 :
+        seconds = days * 86400
+        timestamp = now() + seconds
+    return str(timestamp)
+
 
 
 def save(code, filepath) :

src/mur/admin.py

@@ -15,21 +15,16 @@ def read_machine_code() :
     return read(MACHINE_CODE_PATH)
 
 
-def gen_user_code(days=30, to_file=True) :
+def gen_user_code(days=3, crypt=CRYPT, to_file=True) :
     '''
-    管理员场景： 随机分配用户码，并写入文件
+    管理员场景： 分配用户码，并写入文件
     （可直接把文件发送给用户，让其放到程序根目录）
     [param] day: 授权天数。 0 为永久授权
     [param] to_file: 是否把用户码写入文件
     [return] 用户码
     '''
-    str_range = "%s%s" % (
-        string.ascii_letters, 
-        string.digits
-    )
-    user_code = ''.join(
-        random.sample(str_range, 8)
-    )
+    after_time = after(int(days))
+    user_code = crypt.encrypt_des(after_time)
     if to_file :
         save(user_code, USER_CODE_PATH)
     return user_code
@@ -51,7 +46,8 @@ def gen_register_code(machine_code, user_code, crypt=CRYPT, to_file=True) :
     '''
     try :
         uuid = crypt.decrypt_des(machine_code)
-        register_code = gen_rc(crypt, uuid, user_code)
+        expire_time = crypt.decrypt_des(user_code)
+        register_code = gen_rc(crypt, uuid, expire_time)
     except :
         register_code = ''
         print('无法解密【机器码】：加密密钥不同、或加密格式不正确')

src/mur/user.py

@@ -11,7 +11,7 @@ def gen_machine_code(crypt=CRYPT, to_file=True) :
     （可直接把文件发送给管理员，让其生成注册码）
     [param] crypt: 加解密类
     [param] to_file: 是否把用户码写入文件
-    [return] 用户码
+    [return] 机器码码
     '''
     uuid = MI.generate()
     machine_code = crypt.encrypt_des(uuid)
@@ -20,19 +20,18 @@ def gen_machine_code(crypt=CRYPT, to_file=True) :
     return machine_code
 
 
-def read_user_code(tips='请输入用户码: ') :
+def read_user_code() :
     '''
     用户场景： 读取（或输入）被管理员分配的用户码
-    [param] tips: 输入提示
     [return] 用户码
     '''
     user_code = read(USER_CODE_PATH)   # 若无法从文件中读取
     if user_code == '' :
-        user_code = input(tips)   # 则要求用户输入
+        user_code = input('请输入用户码: ')   # 则要求用户输入
     return user_code
 
 
-def verify_authorization(user_code, crypt=CRYPT, tips='用户码错误 或 注册码不存在，请联系管理员。程序终止。') :
+def verify_authorization(user_code, crypt=CRYPT) :
     '''
     用户场景： 每次运行程序时，
         1. 输入用户码 
@@ -42,12 +41,23 @@ def verify_authorization(user_code, crypt=CRYPT, tips='用户码错误 或 注
         5. 比较两个注册码是否相同
     [param] user_code: 用户码
     [param] crypt: 加解密类
-    [param] tips: 错误提示
     [return] true: 注册码一致； false: 注册码不同
     '''
     uuid = MI.generate()
-    register_code = gen_rc(crypt, uuid, user_code)
+    expire_time = crypt.decrypt_des(user_code)
+    register_code = gen_rc(crypt, uuid, expire_time)
     rst = (register_code == read(REGISTER_CODE_PATH))
-    if not rst :
-        print(tips)
+    if rst :
+        expire_time = int(expire_time)
+        expire_date = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(expire_time))
+        if expire_time == 0 :
+            print('注册码正确（永久）。')
+
+        elif now() <= expire_time :
+            print('注册码正确（有效期至 %s）。' % expire_date)
+        else :
+            rst = False
+            print('注册码已过期（有效期至 %s）。' % expire_date)
+    else :
+        print('用户码错误 或 注册码不存在，请联系管理员。')
     return rst

tests/test.py

@@ -12,6 +12,7 @@
 # ----------------------------------------------------------------------
 
 import unittest
+import time
 from src.mur.crypt import *
 from src.mur.admin import *
 from src.mur.user import *
@@ -62,14 +63,14 @@ def test_gen_machine_code(self) :
         )
 
 
-    def test_gen_user_code(self) :
-        BIT = 16
-        user_code = gen_user_code(BIT)
-        print("用户码： %s" % user_code)
-        self.assertEqual(
-            len(user_code), 
-            BIT
-        )
+    # def test_gen_user_code(self) :
+    #     DAYS = 30
+    #     user_code = gen_user_code(DAYS)
+    #     print("用户码： %s" % user_code)
+    #     self.assertEqual(
+    #         int(self.CRYPT.decrypt_des(user_code)), 
+    #         (int(time.time()) + 86400 * DAYS)
+    #     )
 
 
     def test_gen_register_code(self) :
@@ -79,7 +80,7 @@ def test_gen_register_code(self) :
 
         machine_code = gen_machine_code(my_crypt)
         print("机器码： %s" % machine_code)
-        user_code = gen_user_code()
+        user_code = gen_user_code(0, my_crypt)
         print("用户码： %s" % user_code)
         register_code = gen_register_code(machine_code, user_code, my_crypt)
         print("注册码： %s" % register_code)
@@ -101,7 +102,8 @@ def test_verify_authorization(self) :
         #--------------------
         # 管理员本地执行：
         a_machine_code = read_machine_code()    # 管理员读取用户提供的机器码文件
-        a_user_code = gen_user_code()           # 管理员随机分配的用户码（会生成文件）
+        days = 5                                # 授权天数
+        a_user_code = gen_user_code(days)       # 管理员随机分配的用户码（会生成文件）
         a_register_code = gen_register_code(    # 管理员为用户生成的注册码（会生成文件）
             a_machine_code, a_user_code
         )