Python 脚本如何设置试用期

程序员可能会私下给别人开发一些工具，但是通常要给别人试用一下，但是万一别人试用后，把你拉黑，那就白忙活了。今天就分享如何在 Python 里设置有效期。

为了设置有效期，首先要获取当前时间，如果获取本地时间，那么别人可以通过修改操作系统的时间来一直保持有效。最好的办法就是获取网络时间。你说他断网怎么办？

如果断网，有两种方法，要么不给用，要么再取本地时间。

1、获取网络时间

如何获取网络时间呢，可以从 HTTP 返回的 Header 里获取时间，由于这个时间是 GMT 格式的 UTC 时间，我们还需要转为本地时间，下面的脚本都考虑到了，均采用标准库。

from datetime import datetime
from urllib.request import urlopen
import time
import ssl
import sys

ssl._create_default_https_context = ssl._create_unverified_context


def get_network_time():
    url = "https://www.baidu.com"
    try:
        with urlopen(url) as res:
            utctime = gmtstr_to_localtime(res.getheader("Date"))
            return datetime_from_utc_to_local(utctime)
    except Exception as e:
        print("请检查网络设置")
        sys.exit()


def gmtstr_to_localtime(gmtstr) -> datetime:
    return datetime.strptime(gmtstr, "%a, %d %b %Y %H:%M:%S GMT")


def datetime_from_utc_to_local(utc_datetime):
    now_timestamp = time.time()
    offset = datetime.fromtimestamp(now_timestamp) - datetime.utcfromtimestamp(
        now_timestamp
    )
    return utc_datetime + offset


if __name__ == "__main__":
    print(get_network_time())
    # Output:
    # 2022-07-19 06:39:12

2、上锁

这一步很简单，一个日期判断就搞定了：

def lock():
    now = get_network_time()
    end_date = datetime(2022, 8, 26)
    if now > end_date:
        print("试用已到期")
        sys.exit()

然后把上面获取网络时间的代码及 lock 函数均放在 lock.py 文件里，在我们的核心模块 core_work.py 中导入：

from lock import lock

def somefunc():
    lock()
    # do your work

在需要 lock() 的地方调用下 lock 函数就可以了。

你可能还需要一个 startup.py 脚本来启动整个程序：

from core_work import main
main()

3、编译 pyd

不编译的话，或者编译为 pyc 的话，上面的工作等于没做，pyc 可以直接反编译成 py 文件，连命名符号都不变。

编译的话，只需要需要把获取网络时间的代码，lock 函数的定义及调用 lock 的代码都编译成 pyd，本例子中就是 lock.py 和 core_work.py 编译成 lock.pyd 和 core_work.pyd,然后把源代码 lock.py 和 core_work.py 删除，其实就可以发布了，想破解只能反编译 pyd 文件，得到的是汇编代码，这个门槛已经拒绝了大多数的人。

pyd 的编译流程是 .py -> .c -> pyd，也就是说 pyd 是从 c 语言编译的，除了隐藏源代码之外，速度也会变快一些。

如果不放心的话，那就 pyinstall -k key startup.py 再加密打包一下，key 就是加密的密码，如果在 pyd 文件 import 了第三方库，那么要在 spec 文件中填写 hiddenimports。

如何编译 pyd 呢？代码如下：

from distutils.core import setup
from Cython.Build import cythonize
files = ["lock.py", "core_work.py"]

setup(
    name="yourapp",
    ext_modules=cythonize(files),
    script_args=["build_ext", "-b", "./build", "-t", "./build/temp"],
)

运行之前你要 pip install cython。

然后在 build 文件夹就可以看到编译好的 pyd 文件，然后把他们移动到源代码的位置，重命名为 xxx.pyd 就可以了。

最后的话

以上编程技巧，简单实用，如果有帮助，还请点赞、在看、转发支持。有问题，欢迎留言讨论。