python-gitlab/gitlab/utils.py at main · d3alxi/python-gitlab · GitHub

Repository navigation

History

88 lines (68 loc) · 2.81 KB

Raw

# -*- coding: utf-8 -*-

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public License

# along with this program. If not, see <http://www.gnu.org/licenses/>.

import urllib.parse

from typing import Any, Callable, Dict, Optional, Union

import requests

class _StdoutStream:

def __call__(self, chunk: Any) -> None:

print(chunk)

def response_content(

response: requests.Response,

streamed: bool,

action: Optional[Callable],

chunk_size: int,

) -> Optional[bytes]:

if streamed is False:

return response.content

if action is None:

action = _StdoutStream()

for chunk in response.iter_content(chunk_size=chunk_size):

if chunk:

action(chunk)

return None

def copy_dict(dest: Dict[str, Any], src: Dict[str, Any]) -> None:

for k, v in src.items():

if isinstance(v, dict):

# Transform dict values to new attributes. For example:

# custom_attributes: {'foo', 'bar'} =>

# "custom_attributes['foo']": "bar"

for dict_k, dict_v in v.items():

dest[f"{k}[{dict_k}]"] = dict_v

else:

dest[k] = v

class EncodedId(str):

"""A custom `str` class that will return the URL-encoded value of the string.

* Using it recursively will only url-encode the value once.

* Can accept either `str` or `int` as input value.

* Can be used in an f-string and output the URL-encoded string.

Reference to documentation on why this is necessary.

See::

https://docs.gitlab.com/ee/api/index.html#namespaced-path-encoding

https://docs.gitlab.com/ee/api/index.html#path-parameters

"""

# mypy complains if return type other than the class type. So we ignore issue.

def __new__( # type: ignore

cls, value: Union[str, int, "EncodedId"]

) -> Union[int, "EncodedId"]:

if isinstance(value, (int, EncodedId)):

return value

if not isinstance(value, str):

raise TypeError(f"Unsupported type received: {type(value)}")

value = urllib.parse.quote(value, safe="")

return super().__new__(cls, value)

def remove_none_from_dict(data: Dict[str, Any]) -> Dict[str, Any]:

return {k: v for k, v in data.items() if v is not None}

python-gitlab/gitlab/utils.py at main · d3alxi/python-gitlab · GitHub

Repository navigation

Expand file tree

Breadcrumbs

utils.py

Latest commit

History

Breadcrumbs

utils.py

File metadata and controls