Style with black

This commit is contained in:
Kyle Hornberg
2020-04-13 16:19:27 -05:00
parent 858d4c0b7f
commit ddacfed1ff
2 changed files with 19 additions and 11 deletions
+18 -10
View File
@@ -27,9 +27,13 @@ class Octokit(Base):
class_attributes = defaultdict(dict) class_attributes = defaultdict(dict)
for path, path_object in definitions["paths"].items(): for path, path_object in definitions["paths"].items():
for method, method_object in path_object.items(): for method, method_object in path_object.items():
cls_name, method_id_name = self._get_names_from_operation_id(method_object) cls_name, method_id_name = self._get_names_from_operation_id(
method_object
)
method_name = utils.snake_case(str(method_object.get("summary"))) method_name = utils.snake_case(str(method_object.get("summary")))
methods = self._get_class_methods(method_id_name, method_name, method_object, method, path) methods = self._get_class_methods(
method_id_name, method_name, method_object, method, path
)
class_attributes[cls_name].update(methods) class_attributes[cls_name].update(methods)
return class_attributes return class_attributes
@@ -65,23 +69,27 @@ class Octokit(Base):
def _get_deprecated_methods(self, methods, method_object): def _get_deprecated_methods(self, methods, method_object):
deprecated_methods = {} deprecated_methods = {}
if method_object.get('x-changes'): if method_object.get("x-changes"):
for change in method_object['x-changes']: for change in method_object["x-changes"]:
if change.get("type") == "operation": if change.get("type") == "operation":
before_cls, before_name = self._get_names_from_operation_id(change.get("before")) before_cls, before_name = self._get_names_from_operation_id(
after_cls, after_name = self._get_names_from_operation_id(change.get("after")) change.get("before")
)
after_cls, after_name = self._get_names_from_operation_id(
change.get("after")
)
if before_cls == after_cls and methods.get(after_name): if before_cls == after_cls and methods.get(after_name):
deprecated_methods.update({before_name: methods[after_name]}) deprecated_methods.update({before_name: methods[after_name]})
return deprecated_methods return deprecated_methods
def _get_class_methods(self, method_id_name, method_name, method_object, method, path): def _get_class_methods(
self, method_id_name, method_name, method_object, method, path
):
methods = { methods = {
method_id_name: self._create_method( method_id_name: self._create_method(
method_id_name, method_object, method, path method_id_name, method_object, method, path
), ),
method_name: self._create_method( method_name: self._create_method(method_name, method_object, method, path),
method_name, method_object, method, path
),
} }
methods.update(self._get_deprecated_methods(methods, method_object)) methods.update(self._get_deprecated_methods(methods, method_object))
return methods return methods
+1 -1
View File
@@ -166,7 +166,7 @@ class TestClientMethods(object):
"Content-Type": "application/json", "Content-Type": "application/json",
"accept": "application/vnd.github.v3+json", "accept": "application/vnd.github.v3+json",
} }
data = {'sort': 'updated', 'per_page': 30, 'page': 1} data = {"sort": "updated", "per_page": 30, "page": 1}
Octokit().issues.list_comments_for_repo( Octokit().issues.list_comments_for_repo(
owner="testUser", repo="testRepo", **data owner="testUser", repo="testRepo", **data
) )