github/pipreqs
1
1
Fork 0
mirror of https://github.com/bndr/pipreqs.git synced 2025-06-06 03:25:21 +00:00
Code Issues 3 Packages Projects Releases Wiki Activity

str.partition() simplifies the logic, thighten up write()

Browse Source 
str.partition() has nice advantages over str.split():

```python
to_append = item if "." not in item else item.split(".")[0]
# is equivalent to...
to_append = item.partition(".")[0]
```

Also tightened up generate_requirements_file() to send all data in a single write() operation
This commit is contained in:
cclauss 2015-04-28 07:35:54 +02:00
parent 7ee7e75358
commit 0b4b522c38
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pipreqs/pipreqs.py
View File

@ -51,11 +51,8 @@ def get_all_imports(start_path):
if "," in item:
for match in item.split(","):
imports.append(match.strip())
elif " as " in item:
to_append = item.split(" as ")[0]
imports.append(to_append.strip())
else:
to_append = item if "." not in item else item.split(".")[0]
to_append = item.partition(' as ')[0].partition('.')[0]
imports.append(to_append.strip())
third_party_packages = set(imports) - set(set(packages) & set(imports))
logging.debug('Found third-party packages: %s', third_party_packages)
@ -65,11 +62,10 @@ def get_all_imports(start_path):
def generate_requirements_file(path, imports):
with open(path, "w") as ff:
logging.debug('Writing requirements to file %s', path)
for item in imports:
ff.write(item['name'] + "==" + item['version'])
ff.write("\n")
with open(path, "w") as out_file:
logging.debug('Writing %d requirements to file %s', (len(imports), path))
fmt = '{name} == {version}'
out_file.write('\n'.join(fmt.format(**item) for item in imports) + '\n')
def get_imports_info(imports):