-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.py
38 lines (30 loc) · 873 Bytes
/
requirements.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import subprocess
def main():
# generate requirements
process = subprocess.Popen(
['pipreqs', './', '--force'],
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
out, err = process.communicate()
if process.returncode != 0:
print(str(err, 'utf-8'))
return
# settings
requirements_file = 'requirements.txt'
# read
with open(requirements_file, 'r') as requirements:
content = requirements.read().splitlines()
# sort content
content = list(set(content))
content.sort(key=lambda y: y.lower())
content = '\n'.join(content)
# save
with open(requirements_file, 'w') as requirements:
requirements.write(f'{content}\n')
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
print('\nGoodbye!')