glpkg 1.0.0__tar.gz → 1.2.0__tar.gz

{glpkg-1.0.0 → glpkg-1.2.0}/PKG-INFO

@@ -1,25 +1,19 @@
 Metadata-Version: 2.4
 Name: glpkg
-Version: 1.0.0
+Version: 1.2.0
 Summary: Tool to make GitLab generic package registry operations easy.
 Author-email: bugproduction <bugproduction@outlook.com>
 License-Expression: MIT
 Project-URL: Repository, https://gitlab.com/bugproduction/glpkg.git
 Keywords: GitLab,packages,registry,generic
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Provides-Extra: dev
-Requires-Dist: black; extra == "dev"
-Requires-Dist: build; extra == "dev"
-Requires-Dist: pytest; extra == "dev"
-Requires-Dist: pytest-cov; extra == "dev"
-Requires-Dist: setuptools; extra == "dev"
-Requires-Dist: twine; extra == "dev"
 Dynamic: license-file
 
-# glpkg - GitLab Generic Package registry tools 
+# glpkg - GitLab Generic Package tools 
 
-glpkg is a tool that makes it easy to work with [GitLab generic package registry](https://docs.gitlab.com/user/packages/generic_packages/).
+glpkg is a tool that makes it easy to work with [GitLab generic packages](https://docs.gitlab.com/user/packages/generic_packages/).
 
 
 ## Installation
@@ -47,7 +41,11 @@ By default, the used GitLab host is gitlab.com. If you use a self-hosted GitLab,
 
 To authenticate with the package registry in any of the commands below, use `--token readapitoken123` argument where the `readapitoken123` is a [personal](https://docs.gitlab.com/user/profile/personal_access_tokens/#create-a-personal-access-token) or [project](https://docs.gitlab.com/user/project/settings/project_access_tokens/#create-a-project-access-token) access token, with read API scope. In case the package registry is public, you can omit this argument.
 
-The above arguments are omitted in the examples below to focus on the functions. Add the arguments to change the host or to authenticate with the registry.
+Alternatively you can use a token stored in your `.netrc` file by setting `--netrc` argument.
+
+> If you use the tool in GitLab CI, read [below](#Use-in-GitLab-pipelines) on how to use the `CI_JOB_TOKEN`.
+
+The arguments related to the GitLab host or authentication (`--token`, `--netrc`, and `--ci`) are omitted in the examples below to focus on the commands.
 
 In general, run `glpkg --help` when needed.
 
@@ -74,7 +72,7 @@ mypackagename   2.0
 
 ### Download generic package
 
-To download everything from a specific generic package, run
+To download all files from a specific version of a generic package, run
 
 ```bash
 glpkg download --project 12345 --name mypackagename --version 1.0
@@ -85,7 +83,13 @@ Where:
 - `mypackagename` is the name of the generic package
 - `1.0` is the version of the generic package from which the files are downloaded
 
-The files will be downloaded in the current working directory. Any pre-existing files will be overridden without warning.
+By default the files will be downloaded in the current working directory. To download the files to another directory, add argument `--destination` to the command. In all cases, as long as you have permissions to the destination directory, any pre-existing files will be overridden without warning.
+
+To download only a specific file from the package, add `--file` argument.
+
+```bash
+glpkg download --project 12345 --name mypackagename --version 1.5 --file the_only_one --destination /temp
+```
 
 > If a package has multiple files with the same filename, the tool can only download the newest file. This is a restriction of GitLab API.
 
@@ -107,7 +111,7 @@ Where:
 
 ### Use in GitLab pipelines
 
-If you use the tool in a GitLab pipeline, using `--ci` argument uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID and to use a personal or project access token instead of CI_JOB_TOKEN.
+If you use the tool in a GitLab pipeline, setting argument `--ci` uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID or to use a personal or project access token instead of `CI_JOB_TOKEN`.
 
 In other words, you don't need to give the `--host`, `--project`, or `--token` arguments if you are interacting with the package registry of the project where the pipeline is running. Example: uploading `my-file.txt` to generic package `mypackagename` version `1.0` in the project package registry in CI:
 
@@ -124,4 +128,3 @@ The tool is not perfect (yet) and has limitations. The following limitations are
 
 - Uploading files must be done one-by-one.
 - Only project registries are supported for now.
-- Pagination is not supported for now - in case you have more than 100 versions of a package, not all will be shown.

{glpkg-1.0.0 → glpkg-1.2.0}/README.md

@@ -1,6 +1,6 @@
-# glpkg - GitLab Generic Package registry tools 
+# glpkg - GitLab Generic Package tools 
 
-glpkg is a tool that makes it easy to work with [GitLab generic package registry](https://docs.gitlab.com/user/packages/generic_packages/).
+glpkg is a tool that makes it easy to work with [GitLab generic packages](https://docs.gitlab.com/user/packages/generic_packages/).
 
 
 ## Installation
@@ -28,7 +28,11 @@ By default, the used GitLab host is gitlab.com. If you use a self-hosted GitLab,
 
 To authenticate with the package registry in any of the commands below, use `--token readapitoken123` argument where the `readapitoken123` is a [personal](https://docs.gitlab.com/user/profile/personal_access_tokens/#create-a-personal-access-token) or [project](https://docs.gitlab.com/user/project/settings/project_access_tokens/#create-a-project-access-token) access token, with read API scope. In case the package registry is public, you can omit this argument.
 
-The above arguments are omitted in the examples below to focus on the functions. Add the arguments to change the host or to authenticate with the registry.
+Alternatively you can use a token stored in your `.netrc` file by setting `--netrc` argument.
+
+> If you use the tool in GitLab CI, read [below](#Use-in-GitLab-pipelines) on how to use the `CI_JOB_TOKEN`.
+
+The arguments related to the GitLab host or authentication (`--token`, `--netrc`, and `--ci`) are omitted in the examples below to focus on the commands.
 
 In general, run `glpkg --help` when needed.
 
@@ -55,7 +59,7 @@ mypackagename   2.0
 
 ### Download generic package
 
-To download everything from a specific generic package, run
+To download all files from a specific version of a generic package, run
 
 ```bash
 glpkg download --project 12345 --name mypackagename --version 1.0
@@ -66,7 +70,13 @@ Where:
 - `mypackagename` is the name of the generic package
 - `1.0` is the version of the generic package from which the files are downloaded
 
-The files will be downloaded in the current working directory. Any pre-existing files will be overridden without warning.
+By default the files will be downloaded in the current working directory. To download the files to another directory, add argument `--destination` to the command. In all cases, as long as you have permissions to the destination directory, any pre-existing files will be overridden without warning.
+
+To download only a specific file from the package, add `--file` argument.
+
+```bash
+glpkg download --project 12345 --name mypackagename --version 1.5 --file the_only_one --destination /temp
+```
 
 > If a package has multiple files with the same filename, the tool can only download the newest file. This is a restriction of GitLab API.
 
@@ -88,7 +98,7 @@ Where:
 
 ### Use in GitLab pipelines
 
-If you use the tool in a GitLab pipeline, using `--ci` argument uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID and to use a personal or project access token instead of CI_JOB_TOKEN.
+If you use the tool in a GitLab pipeline, setting argument `--ci` uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID or to use a personal or project access token instead of `CI_JOB_TOKEN`.
 
 In other words, you don't need to give the `--host`, `--project`, or `--token` arguments if you are interacting with the package registry of the project where the pipeline is running. Example: uploading `my-file.txt` to generic package `mypackagename` version `1.0` in the project package registry in CI:
 
@@ -104,5 +114,4 @@ To use the `CI_JOB_TOKEN` with package registry of another projects, add `--proj
 The tool is not perfect (yet) and has limitations. The following limitations are known, but more can exist:
 
 - Uploading files must be done one-by-one.
-- Only project registries are supported for now.
-- Pagination is not supported for now - in case you have more than 100 versions of a package, not all will be shown.
+- Only project registries are supported for now.

{glpkg-1.0.0 → glpkg-1.2.0}/pyproject.toml

@@ -12,21 +12,12 @@ license-files = ["LICENSE.md"]
 authors = [
   { name="bugproduction", email="bugproduction@outlook.com" }
 ]
+requires-python = ">= 3.9"
 dependencies = []
 
 [project.urls]
 Repository = "https://gitlab.com/bugproduction/glpkg.git"
 
-[project.optional-dependencies]
-dev = [
-    "black",
-    "build",
-    "pytest",
-    "pytest-cov",
-    "setuptools",
-    "twine"
-]
-
 [project.scripts]
 glpkg = "gitlab.__main__:cli"
 

{glpkg-1.0.0 → glpkg-1.2.0}/src/gitlab/__init__.py

@@ -1,3 +1,3 @@
 from gitlab.packages import Packages
 
-__version__ = "1.0.0"
+__version__ = "1.2.0"

{glpkg-1.0.0 → glpkg-1.2.0}/src/gitlab/cli_handler.py

@@ -1,6 +1,8 @@
 import argparse
 import netrc
 import os
+import sys
+import urllib
 from gitlab import Packages, __version__
 
 
@@ -33,7 +35,20 @@ class CLIHandler:
         return 0
 
     def do_it(self) -> int:
-        return self.args.action(self.args)
+        ret = 1
+        try:
+            ret = self.args.action(self.args)
+        except urllib.error.HTTPError as e:
+            # GitLab API returns 404 when a resource is not found
+            # but also when the user has no access to the resource
+            print("Oops! Something did go wrong.", file=sys.stderr)
+            print(e, file=sys.stderr)
+            print(
+                "Note that Error 404 may also indicate authentication issues with GitLab API.",
+                file=sys.stderr,
+            )
+            print("Check your arguments and credentials.", file=sys.stderr)
+        return ret
 
     def _register_common_arguments(self, parser) -> None:
         group = parser.add_mutually_exclusive_group()
@@ -73,6 +88,19 @@ class CLIHandler:
     def _register_download_parser(self, parser):
         self._register_common_arguments(parser)
         parser.add_argument("-v", "--version", type=str, help="The package version.")
+        parser.add_argument(
+            "-f",
+            "--file",
+            type=str,
+            help="The file to download from the package. If not defined, all files are downloaded.",
+        )
+        parser.add_argument(
+            "-d",
+            "--destination",
+            default="",
+            type=str,
+            help="The path where the file(s) are downloaded. If not defined, the current working directory is used.",
+        )
         parser.set_defaults(action=self._download_handler)
 
     def _args(self, args):
@@ -98,16 +126,25 @@ class CLIHandler:
         return host, project, name, token_user, token
 
     def _download_handler(self, args) -> int:
+        ret = 1
         host, project, name, token_user, token = self._args(args)
         version = args.version
+        destination = args.destination
         gitlab = Packages(host, token_user, token)
         package_id = gitlab.get_package_id(project, name, version)
-        files = gitlab.list_files(project, package_id)
-        ret = 1
-        for file in files:
-            ret = gitlab.download_file(project, name, version, file)
-            if not ret:
-                break
+        if package_id:
+            files = []
+            if args.file:
+                files.append(args.file)
+            else:
+                files = gitlab.list_files(project, package_id)
+            for file in files:
+                ret = gitlab.download_file(project, name, version, file, destination)
+                if ret:
+                    print("Failed to download file " + file)
+                    break
+        else:
+            print("No package " + name + " version " + version + " found!")
         return ret
 
     def _register_list_parser(self, parser):
@@ -134,9 +171,13 @@ class CLIHandler:
         parser.set_defaults(action=self._upload)
 
     def _upload(self, args) -> int:
+        ret = 1
         host, project, name, token_user, token = self._args(args)
         version = args.version
         file = args.file
-        gitlab = Packages(host, token_user, token)
-        ret = gitlab.upload_file(project, name, version, file)
+        if os.path.isfile(file):
+            gitlab = Packages(host, token_user, token)
+            ret = gitlab.upload_file(project, name, version, file)
+        else:
+            print("File " + file + " does not exist!")
         return ret

glpkg-1.2.0/src/gitlab/packages.py

@@ -0,0 +1,176 @@
+from http.client import HTTPMessage
+import json
+import logging
+import os
+from urllib import request, parse
+
+logger = logging.getLogger(__name__)
+
+
+class Packages:
+    def __init__(self, host: str, token_type: str, token: str):
+        self.host = host
+        self.token_type = token_type
+        self.token = token
+
+    def api_url(self) -> str:
+        return "https://{}/api/v4/".format(parse.quote(self.host))
+
+    def project_api_url(self, project: str) -> str:
+        return self.api_url() + "projects/{}/".format(parse.quote_plus(project))
+
+    def get_headers(self) -> dict:
+        headers = {}
+        if self.token_type and self.token:
+            headers = {self.token_type: self.token}
+        return headers
+
+    def _request(self, url: str) -> tuple[int, bytes, HTTPMessage]:
+        logger.debug("Requesting " + url)
+        req = request.Request(url, headers=self.get_headers())
+        with request.urlopen(req) as response:
+            return response.status, response.read(), response.headers
+
+    def _get_next_page(self, headers: HTTPMessage) -> int:
+        ret = 0
+        if headers:
+            next_page = headers.get("x-next-page")
+            if next_page:
+                ret = int(next_page)
+                logger.debug("Response incomplete, next page is " + next_page)
+            else:
+                logger.debug("Response complete")
+        return ret
+
+    def _build_query(self, arg: str, page: int) -> str:
+        query = ""
+        if arg or page:
+            if page:
+                page = "page=" + str(page)
+            query = "?{}".format("&".join(filter(None, (arg, page))))
+        return query
+
+    def gl_project_api(self, project: str, path: str, arg: str = None) -> list:
+        data = []
+        more = True
+        page = None
+        while more:
+            more = False
+            query = self._build_query(arg, page)
+            url = self.project_api_url(project) + path + query
+            status, res_data, headers = self._request(url)
+            logger.debug("Response status: " + str(status))
+            res_data = json.loads(res_data)
+            logger.debug("Response data: " + str(res_data))
+            data = data + res_data
+            page = self._get_next_page(headers)
+            if page:
+                more = True
+        return data
+
+    def list_packages(self, project: str, package_name: str) -> list:
+        packages = []
+        logger.debug("Listing packages with name " + package_name)
+        data = self.gl_project_api(
+            project, "packages", "package_name=" + parse.quote_plus(package_name)
+        )
+        for package in data:
+            name = parse.unquote(package["name"])
+            version = parse.unquote(package["version"])
+            # GitLab API returns packages that have some match to the filter;
+            # let's filter out non-exact matches
+            if package_name != name:
+                continue
+            packages.append({"name": name, "version": version})
+        return packages
+
+    def list_files(self, project: str, package_id: int) -> list:
+        files = []
+        logger.debug("Listing package " + str(package_id) + " files")
+        path = "packages/" + parse.quote_plus(str(package_id)) + "/package_files"
+        data = self.gl_project_api(project, path)
+        for package in data:
+            # Only append the filename once to the list of files
+            # as there's no way to download them separately through
+            # the API
+            filename = parse.unquote(package["file_name"])
+            if filename not in files:
+                files.append(filename)
+        return files
+
+    def get_package_id(
+        self, project: str, package_name: str, package_version: str
+    ) -> int:
+        id = 0
+        logger.debug(
+            "Fetching package " + package_name + " (" + package_version + ") ID"
+        )
+        path = "packages"
+        arg = (
+            "package_name="
+            + parse.quote_plus(package_name)
+            + "&package_version="
+            + parse.quote_plus(package_version)
+        )
+        data = self.gl_project_api(project, path, arg)
+        if len(data) == 1:
+            package = data.pop()
+            id = package["id"]
+        return id
+
+    def download_file(
+        self,
+        project: str,
+        package_name: str,
+        package_version: str,
+        filename: str,
+        destination: str = "",
+    ) -> int:
+        ret = 1
+        logger.debug("Downloading file " + filename)
+        url = (
+            self.project_api_url(project)
+            + "packages/generic/"
+            + parse.quote_plus(package_name)
+            + "/"
+            + parse.quote_plus(package_version)
+            + "/"
+            + parse.quote(filename)
+        )
+        status, data, _ = self._request(url)
+        if status == 200:
+            path = os.path.join(destination, filename)
+            parent = os.path.dirname(path)
+            if parent:
+                # Create missing directories if needed
+                # In case path has no parent, current
+                # workind directory is used
+                os.makedirs(os.path.dirname(path), exist_ok=True)
+            with open(path, "wb") as file:
+                file.write(data)
+                ret = 0
+        return ret
+
+    def upload_file(
+        self, project: str, package_name: str, package_version: str, file: str
+    ) -> int:
+        ret = 1
+        logger.debug("Uploading file " + file)
+        with open(str(file), "rb") as data:
+            url = (
+                self.project_api_url(project)
+                + "packages/generic/"
+                + parse.quote_plus(package_name)
+                + "/"
+                + parse.quote_plus(package_version)
+                + "/"
+                + parse.quote(str(file))
+            )
+            res = request.urlopen(
+                request.Request(
+                    url, method="PUT", data=data, headers=self.get_headers()
+                )
+            )
+            if res.status == 201:  # 201 is created
+                ret = 0
+        return ret

{glpkg-1.0.0 → glpkg-1.2.0}/src/glpkg.egg-info/PKG-INFO

@@ -1,25 +1,19 @@
 Metadata-Version: 2.4
 Name: glpkg
-Version: 1.0.0
+Version: 1.2.0
 Summary: Tool to make GitLab generic package registry operations easy.
 Author-email: bugproduction <bugproduction@outlook.com>
 License-Expression: MIT
 Project-URL: Repository, https://gitlab.com/bugproduction/glpkg.git
 Keywords: GitLab,packages,registry,generic
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Provides-Extra: dev
-Requires-Dist: black; extra == "dev"
-Requires-Dist: build; extra == "dev"
-Requires-Dist: pytest; extra == "dev"
-Requires-Dist: pytest-cov; extra == "dev"
-Requires-Dist: setuptools; extra == "dev"
-Requires-Dist: twine; extra == "dev"
 Dynamic: license-file
 
-# glpkg - GitLab Generic Package registry tools 
+# glpkg - GitLab Generic Package tools 
 
-glpkg is a tool that makes it easy to work with [GitLab generic package registry](https://docs.gitlab.com/user/packages/generic_packages/).
+glpkg is a tool that makes it easy to work with [GitLab generic packages](https://docs.gitlab.com/user/packages/generic_packages/).
 
 
 ## Installation
@@ -47,7 +41,11 @@ By default, the used GitLab host is gitlab.com. If you use a self-hosted GitLab,
 
 To authenticate with the package registry in any of the commands below, use `--token readapitoken123` argument where the `readapitoken123` is a [personal](https://docs.gitlab.com/user/profile/personal_access_tokens/#create-a-personal-access-token) or [project](https://docs.gitlab.com/user/project/settings/project_access_tokens/#create-a-project-access-token) access token, with read API scope. In case the package registry is public, you can omit this argument.
 
-The above arguments are omitted in the examples below to focus on the functions. Add the arguments to change the host or to authenticate with the registry.
+Alternatively you can use a token stored in your `.netrc` file by setting `--netrc` argument.
+
+> If you use the tool in GitLab CI, read [below](#Use-in-GitLab-pipelines) on how to use the `CI_JOB_TOKEN`.
+
+The arguments related to the GitLab host or authentication (`--token`, `--netrc`, and `--ci`) are omitted in the examples below to focus on the commands.
 
 In general, run `glpkg --help` when needed.
 
@@ -74,7 +72,7 @@ mypackagename   2.0
 
 ### Download generic package
 
-To download everything from a specific generic package, run
+To download all files from a specific version of a generic package, run
 
 ```bash
 glpkg download --project 12345 --name mypackagename --version 1.0
@@ -85,7 +83,13 @@ Where:
 - `mypackagename` is the name of the generic package
 - `1.0` is the version of the generic package from which the files are downloaded
 
-The files will be downloaded in the current working directory. Any pre-existing files will be overridden without warning.
+By default the files will be downloaded in the current working directory. To download the files to another directory, add argument `--destination` to the command. In all cases, as long as you have permissions to the destination directory, any pre-existing files will be overridden without warning.
+
+To download only a specific file from the package, add `--file` argument.
+
+```bash
+glpkg download --project 12345 --name mypackagename --version 1.5 --file the_only_one --destination /temp
+```
 
 > If a package has multiple files with the same filename, the tool can only download the newest file. This is a restriction of GitLab API.
 
@@ -107,7 +111,7 @@ Where:
 
 ### Use in GitLab pipelines
 
-If you use the tool in a GitLab pipeline, using `--ci` argument uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID and to use a personal or project access token instead of CI_JOB_TOKEN.
+If you use the tool in a GitLab pipeline, setting argument `--ci` uses [GitLab predefined variables](https://docs.gitlab.com/ci/variables/predefined_variables/) to configure the tool. In this case `CI_SERVER_HOST`, `CI_PROJECT_ID`, and `CI_JOB_TOKEN` environment variables are used. The `--project`, and `--token` arguments can still be used to override the project ID or to use a personal or project access token instead of `CI_JOB_TOKEN`.
 
 In other words, you don't need to give the `--host`, `--project`, or `--token` arguments if you are interacting with the package registry of the project where the pipeline is running. Example: uploading `my-file.txt` to generic package `mypackagename` version `1.0` in the project package registry in CI:
 
@@ -124,4 +128,3 @@ The tool is not perfect (yet) and has limitations. The following limitations are
 
 - Uploading files must be done one-by-one.
 - Only project registries are supported for now.
-- Pagination is not supported for now - in case you have more than 100 versions of a package, not all will be shown.

{glpkg-1.0.0 → glpkg-1.2.0}/src/glpkg.egg-info/SOURCES.txt

@@ -9,6 +9,6 @@ src/glpkg.egg-info/PKG-INFO
 src/glpkg.egg-info/SOURCES.txt
 src/glpkg.egg-info/dependency_links.txt
 src/glpkg.egg-info/entry_points.txt
-src/glpkg.egg-info/requires.txt
 src/glpkg.egg-info/top_level.txt
-test/test_gitlab.py
+test/test_cli_handler.py
+test/test_packages.py

glpkg-1.2.0/test/test_cli_handler.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch
+
+import urllib
+
+from gitlab import __version__
+from gitlab.cli_handler import CLIHandler
+import sys
+
+from utils import mock_empty_response, mock_one_response
+
+
+class TestCliHandler:
+
+    def test_version(self, capsys):
+        args = ["glpkg", "-v"]
+        with patch.object(sys, "argv", args):
+            handler = CLIHandler()
+            handler.do_it()
+            out, err = capsys.readouterr()
+            assert out == __version__ + "\n"
+            assert err == ""
+
+    def test_list_empty(self, mock_empty_response, capsys):
+        args = ["glpkg", "list", "--project", "18105942", "--name", "AABCComponent"]
+        with patch.object(sys, "argv", args):
+            with patch.object(
+                urllib.request, "urlopen", return_value=mock_empty_response
+            ):
+                handler = CLIHandler()
+                handler.do_it()
+                out, err = capsys.readouterr()
+                assert out == "Name\t\tVersion\n"
+                assert err == ""
+
+    def test_list_one(self, mock_one_response, capsys):
+        args = ["glpkg", "list", "--project", "18105942", "--name", "ABCComponent"]
+        with patch.object(sys, "argv", args):
+            with patch.object(
+                urllib.request, "urlopen", return_value=mock_one_response
+            ):
+                handler = CLIHandler()
+                handler.do_it()
+                out, err = capsys.readouterr()
+                assert out == "Name\t\tVersion\nABCComponent\t0.0.1\n"
+                assert err == ""

glpkg-1.0.0/test/test_gitlab.py → glpkg-1.2.0/test/test_packages.py

@@ -1,14 +1,17 @@
-import io
-import pytest
 import urllib
 from gitlab.packages import *
 from unittest.mock import mock_open, patch
 
+from utils import (
+    ResponseMock,
+    test_gitlab,
+    mock_empty_response,
+    mock_one_response,
+    mock_five_response,
+)
 
-class TestGitLab:
-    @pytest.fixture
-    def test_gitlab(self):
-        return Packages("gl-host", "token-name", "token-value")
+
+class TestPackages:
 
     def test_api_url(self, test_gitlab):
         assert test_gitlab.api_url() == "https://gl-host/api/v4/"
@@ -33,79 +36,72 @@ class TestGitLab:
         test_gitlab = Packages("gl-host", "", "")
         assert test_gitlab.get_headers() == {}
 
-    def test_list_packages_none(self, test_gitlab):
-        data = io.StringIO("[]")
-        with patch.object(urllib.request, "urlopen", return_value=data):
+    def test_list_packages_none(self, test_gitlab, mock_empty_response):
+        with patch.object(urllib.request, "urlopen", return_value=mock_empty_response):
             packages = test_gitlab.list_packages("24", "package-name")
             assert len(packages) == 0
 
-    def test_list_packages_one(self, test_gitlab):
-        data = io.StringIO('[{"name": "package-name", "version": "0.1.2"}]')
-        with patch.object(urllib.request, "urlopen", return_value=data):
-            packages = test_gitlab.list_packages("24", "package-name")
+    def test_list_packages_one(self, test_gitlab, mock_one_response):
+        with patch.object(urllib.request, "urlopen", return_value=mock_one_response):
+            packages = test_gitlab.list_packages("18105942", "ABCComponent")
             assert len(packages) == 1
 
     def test_list_name_packages_filter(self, test_gitlab):
-        data = io.StringIO(
-            '[{"name": "package-name", "version": "0.1.2"}, {"name": "package-name-something", "version": "0.1.2"}]'
+        data = ResponseMock(
+            200,
+            "",
+            '[{"name": "package-name", "version": "0.1.2"}, {"name": "package-name-something", "version": "0.1.2"}]',
         )
         with patch.object(urllib.request, "urlopen", return_value=data):
             packages = test_gitlab.list_packages("24", "package-name")
             assert len(packages) == 1
 
-    def test_list_name_packages_five(self, test_gitlab):
-        data = io.StringIO(
-            '[{"name": "package-name", "version": "0.1"}, {"name": "package-name", "version": "0.2"}, {"name": "package-name", "version": "0.3"}, {"name": "package-name", "version": "0.4"}, {"name": "package-name", "version": "0.5"}]'
-        )
-        with patch.object(urllib.request, "urlopen", return_value=data):
-            packages = test_gitlab.list_packages("24", "package-name")
+    def test_list_name_packages_five(self, test_gitlab, mock_five_response):
+        with patch.object(urllib.request, "urlopen", return_value=mock_five_response):
+            packages = test_gitlab.list_packages("18105942", "ABCComponent")
             assert len(packages) == 5
 
-    def test_list_files_none(self, test_gitlab):
-        data = io.StringIO("[]")
-        with patch.object(urllib.request, "urlopen", return_value=data):
+    def test_list_files_none(self, test_gitlab, mock_empty_response):
+        with patch.object(urllib.request, "urlopen", return_value=mock_empty_response):
             packages = test_gitlab.list_files("24", "123")
             assert len(packages) == 0
 
     def test_list_files_one(self, test_gitlab):
-        data = io.StringIO('[{"file_name": "filea.txt"}]')
+        data = ResponseMock(200, "", '[{"file_name": "filea.txt"}]')
         with patch.object(urllib.request, "urlopen", return_value=data):
             packages = test_gitlab.list_files("24", "123")
             assert len(packages) == 1
 
     def test_list_files_five(self, test_gitlab):
-        data = io.StringIO(
-            '[{"file_name": "filea.txt"}, {"file_name": "fileb.txt"}, {"file_name": "filec.txt"}, {"file_name": "filed.txt"}, {"file_name": "filee.txt"}]'
+        data = ResponseMock(
+            200,
+            "",
+            '[{"file_name": "filea.txt"}, {"file_name": "fileb.txt"}, {"file_name": "filec.txt"}, {"file_name": "filed.txt"}, {"file_name": "filee.txt"}]',
         )
         with patch.object(urllib.request, "urlopen", return_value=data):
             packages = test_gitlab.list_files("24", "123")
             assert len(packages) == 5
 
-    def test_package_id_none(self, test_gitlab):
-        data = io.StringIO("[]")
-        with patch.object(urllib.request, "urlopen", return_value=data):
+    def test_package_id_none(self, test_gitlab, mock_empty_response):
+        with patch.object(urllib.request, "urlopen", return_value=mock_empty_response):
             packages = test_gitlab.get_package_id("24", "package-name", "0.1")
             assert packages == 0
 
     def test_package_id_one(self, test_gitlab):
-        data = io.StringIO('[{"id": 123}]')
+        data = ResponseMock(200, "", '[{"id": 123}]')
         with patch.object(urllib.request, "urlopen", return_value=data):
             packages = test_gitlab.get_package_id("24", "package-name", "0.1")
             assert packages == 123
 
     def test_upload_file(self, test_gitlab):
-        class rmock:
-            def getcode():
-                return 201
-
+        data = ResponseMock(201, "", "[]")
         with patch("builtins.open", mock_open(read_data="data")):
-            with patch.object(urllib.request, "urlopen", return_value=rmock):
+            with patch.object(urllib.request, "urlopen", return_value=data):
                 packages = test_gitlab.upload_file("24", "package-name", "0.1", "file")
                 assert packages == 0
 
     def test_download_file(self, test_gitlab):
-        data = io.StringIO("file-content")
-        m = mock_open()
+        data = ResponseMock(200, "", "file-content")
         with patch("builtins.open", mock_open()) as file_mock:
             # mock_open.write.return_value = 0
             with patch.object(urllib.request, "urlopen", return_value=data):

glpkg-1.0.0/src/gitlab/packages.py

@@ -1,130 +0,0 @@
-import json
-from urllib import request, parse
-
-
-class Packages:
-    def __init__(self, host: str, token_type: str, token: str):
-        self.host = host
-        self.token_type = token_type
-        self.token = token
-
-    def api_url(self) -> str:
-        return "https://{}/api/v4/".format(parse.quote(self.host))
-
-    def project_api_url(self, project: str) -> str:
-        return self.api_url() + "projects/{}/".format(parse.quote_plus(project))
-
-    def get_headers(self):
-        headers = {}
-        if self.token_type and self.token:
-            headers = {self.token_type: self.token}
-        return headers
-
-    def list_packages(self, project: str, package_name: str) -> list:
-        packages = []
-        with request.urlopen(
-            request.Request(
-                self.project_api_url(project)
-                + "packages?package_name="
-                + parse.quote_plus(package_name),
-                headers=self.get_headers(),
-            )
-        ) as res:
-            data = res.read()
-            for package in json.loads(data):
-                name = parse.unquote(package["name"])
-                version = parse.unquote(package["version"])
-                # The GitLab API returns packages that have some match to the filter;
-                # let's filter out non-exact matches
-                if package_name != name:
-                    continue
-                packages.append({"name": name, "version": version})
-        return packages
-
-    def list_files(self, project: str, package_id: int) -> list:
-        files = []
-        with request.urlopen(
-            request.Request(
-                self.project_api_url(project)
-                + "packages/"
-                + parse.quote_plus(str(package_id))
-                + "/package_files",
-                headers=self.get_headers(),
-            )
-        ) as x:
-            data = x.read()
-            for package in json.loads(
-                data,
-            ):
-                # Only append the filename once to the list of files
-                # as there's no way to download them separately through
-                # the API
-                filename = parse.unquote(package["file_name"])
-                if filename not in files:
-                    files.append(filename)
-        return files
-
-    def get_package_id(
-        self, project: str, package_name: str, package_version: str
-    ) -> int:
-        id = 0
-        with request.urlopen(
-            request.Request(
-                self.project_api_url(project)
-                + "packages?package_name="
-                + parse.quote_plus(package_name)
-                + "&package_version="
-                + parse.quote_plus(package_version),
-                headers=self.get_headers(),
-            )
-        ) as res:
-            data = res.read()
-            package = json.loads(data)
-            if len(package) == 1:
-                package = package.pop()
-                id = package["id"]
-        return id
-
-    def download_file(
-        self, project: str, package_name: str, package_version: str, file: str
-    ) -> int:
-        ret = 1
-        with request.urlopen(
-            request.Request(
-                self.project_api_url(project)
-                + "packages/generic/"
-                + parse.quote_plus(package_name)
-                + "/"
-                + parse.quote_plus(package_version)
-                + "/"
-                + parse.quote(str(file)),
-                headers=self.get_headers(),
-            )
-        ) as req:
-            with open(str(file), "wb") as file:
-                file.write(req.read())
-                ret = 0
-        return ret
-
-    def upload_file(
-        self, project: str, package_name: str, package_version: str, file: str
-    ) -> int:
-        ret = 1
-        with open(str(file), "rb") as data:
-            res = request.urlopen(
-                request.Request(
-                    self.project_api_url(project)
-                    + "packages/generic/"
-                    + parse.quote_plus(package_name)
-                    + "/"
-                    + parse.quote_plus(package_version)
-                    + "/"
-                    + parse.quote(str(file)),
-                    method="PUT",
-                    data=data,
-                    headers=self.get_headers(),
-                )
-            )
-            if res.getcode() == 201:  # 201 is created
-                ret = 0
-        return ret

glpkg-1.0.0/src/glpkg.egg-info/requires.txt

@@ -1,8 +0,0 @@
-
-[dev]
-black
-build
-pytest
-pytest-cov
-setuptools
-twine