1
|
|
|
# Copyright Pincer 2021-Present |
|
|
|
|
2
|
|
|
# Full MIT License can be found in `LICENSE` at the project root. |
3
|
|
|
|
4
|
|
|
from __future__ import annotations |
5
|
|
|
|
6
|
|
|
from inspect import getfullargspec |
|
|
|
|
7
|
|
|
from typing import TYPE_CHECKING |
8
|
|
|
|
9
|
|
|
from .types import T, MISSING |
|
|
|
|
10
|
|
|
|
11
|
|
|
if TYPE_CHECKING: |
12
|
|
|
from ..client import Client |
13
|
|
|
from typing import Any, Callable, Dict, List, Optional, Set, Union, Tuple |
14
|
|
|
|
15
|
|
|
|
16
|
|
|
def construct_client_dict(client: Client, data: Dict) -> Dict: |
17
|
|
|
# TODO: fix docs |
|
|
|
|
18
|
|
|
""" |
19
|
|
|
|
20
|
|
|
Parameters |
21
|
|
|
---------- |
22
|
|
|
client |
23
|
|
|
data |
24
|
|
|
|
25
|
|
|
Returns |
26
|
|
|
------- |
27
|
|
|
|
28
|
|
|
""" |
29
|
|
|
return {**data, "_client": client} |
30
|
|
|
|
31
|
|
|
|
32
|
|
|
def remove_none(obj: Union[List, Dict, Set, Tuple]) -> Union[List, Dict, Set, Tuple]: |
|
|
|
|
33
|
|
|
""" |
34
|
|
|
Removes all ``None`` values from a list, dict or set. |
35
|
|
|
|
36
|
|
|
Parameters |
37
|
|
|
---------- |
38
|
|
|
obj : Union[List, Dict, Set, Tuple] |
39
|
|
|
The list, dict, set or tuple to remove ``None`` values from. |
40
|
|
|
|
41
|
|
|
Returns |
42
|
|
|
------- |
43
|
|
|
Union[List, Dict, Set, Tuple] |
44
|
|
|
The list, dict, set or tuple, without ``None`` values. |
45
|
|
|
""" |
46
|
|
|
if isinstance(obj, list): |
|
|
|
|
47
|
|
|
return [i for i in obj if i is not None] |
48
|
|
|
elif isinstance(obj, tuple): |
49
|
|
|
return tuple(i for i in obj if i is not None) |
50
|
|
|
elif isinstance(obj, set): |
51
|
|
|
return obj - {None} |
52
|
|
|
elif isinstance(obj, dict): |
53
|
|
|
return {k: v for k, v in obj.items() if None not in (k, v)} |
54
|
|
|
|
55
|
|
|
|
56
|
|
|
def dict_to_query(data: Dict[str, Any]) -> str: |
57
|
|
|
""" |
58
|
|
|
Takes a dictionary of arguments and converts it into a query string to append to a url. |
59
|
|
|
|
60
|
|
|
Parameters |
61
|
|
|
---------- |
62
|
|
|
data : Dict[:class:`str`, :class:`~typing.Any`] |
63
|
|
|
The arguments for the query string |
64
|
|
|
|
65
|
|
|
Returns |
66
|
|
|
------- |
67
|
|
|
str |
68
|
|
|
The query string to append to a url |
69
|
|
|
""" |
70
|
|
|
query_str = "" |
71
|
|
|
for key in data: |
72
|
|
|
if data[key] is not None: |
73
|
|
|
query_str += f"{key}={data[key]}&" |
74
|
|
|
return query_str[:-1] |
75
|
|
|
|