python 根据当前工程生成requirements.txt

最近在整理工程，发现好多工程写的比较乱，有的连requirements都没有。上网查找主要是用freeze和pipreqs自动生成的，freeze是pip内置的包，用起来很方便，但是freeze会把环境的所有依赖包都存下来。而pipreqs配置了好多遍都不成功，所以就自己写了一个小脚本。原理：首先用freeze生成当前环境的所有依赖，然后通过对工程内所有.py文件内的引用进行抽取，得到另一个临时的工程依赖文件。最后通过工程内用到的依赖和freeze生成的依赖进行对比，生成最终的requirements。使用方法：可以输入文件地址，或者直接按回车，默认当前目录为工程根目录。会在对应的目录下生成requirements.txt