Merge pull request #14 from martincjespersen/main

Poetry lock support

Author: abduhbm

README.md

@@ -17,6 +17,7 @@ Supported dependency files:
 * `Pipfile`
 * `Pipfile.lock`
 * `conda.yaml`
+* `poetry.lock`
 
 ### Installation
 

dep_license/__init__.py

@@ -32,6 +32,7 @@
     "pyproject.toml",
     "setup.py",
     "conda.yml",
+    "poetry.lock",
 ]
 PYPYI_URL = "https://pypi.python.org/pypi"
 COLUMNS = ["Name", "Meta", "Classifier"]
@@ -130,7 +131,7 @@ def worker(d):
             license_class.add("::".join([x.strip() for x in c.split("::")[1:]]))
 
     license_class_str = (
-        license_class.pop() if len(license_class) == 1 else str(license_class)
+        license_class.pop() if len(license_class) == 1 else ", ".join(license_class)
     )
     record.append(license_class_str)
 

dep_license/utils.py

@@ -31,6 +31,8 @@ def parse_file(input_file, base_name, dev=False):
         elif base_name == "conda.yml":
             return parse_conda_yaml_file(input_file)
 
+        elif base_name == "poetry.lock":
+            return parse_poetry_lock_file(input_file)
         else:
             return parse_req_file(input_file)
     except Exception as e:
@@ -93,6 +95,13 @@ def parse_pyproject_file_poetry(input_file):
     ]
 
 
+def parse_poetry_lock_file(input_file):
+    output = []
+    cf = toml.load(input_file)
+    output = [pkg["name"] for pkg in cf.get("package", []) if "name" in pkg]
+    return output
+
+
 def parse_setup_file(input_file):
     output = []
     cur_dir = os.getcwd()

setup.py

@@ -81,6 +81,7 @@ def run(self):
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
         "Topic :: Utilities",
         "Topic :: System :: System Shells",
     ],

tests/test_utils.py

@@ -176,8 +176,40 @@ def test_parsing_conda_yaml_file(tmpdir):
     assert utils.parse_conda_yaml_file(x.strpath) == ["numpy", "pandas"]
 
 
+def test_parsing_poetry_lock_file(tmpdir):
+    x = tmpdir.join("poetry.lock")
+    x.write(
+        """
+        [[package]]
+        name = "anyio"
+        version = "3.5.0"
+        description = "High level compatibility layer for multiple asynchronous event loop implementations"
+        category = "dev"
+        optional = false
+        python-versions = ">=3.6.2"
+
+        [[package]]
+        name = "appnope"
+        version = "0.1.2"
+        description = "Disable App Nap on macOS >= 10.9"
+        category = "dev"
+        optional = false
+        python-versions = "*"
+        """
+    )
+    assert utils.parse_poetry_lock_file(x.strpath) == ["anyio", "appnope"]
+
+
 @pytest.mark.parametrize(
-    "f", ("requirements.txt", "pyproject.toml", "Pipfile", "Pipfile.lock", "conda.yml")
+    "f",
+    (
+        "requirements.txt",
+        "pyproject.toml",
+        "Pipfile",
+        "Pipfile.lock",
+        "conda.yml",
+        "poetry.lock",
+    ),
 )
 def test_passing_dependency_file(f, tmpdir):
     x = tmpdir.join(f)