Refactor code for improved readability and PEP 8 compliance

This commit is contained in:
mahiuddin-dev 2023-10-24 11:47:49 +06:00
parent f56b85b7dd
commit f8b64c4fbb

View File

@ -1,45 +1,44 @@
from operator import attrgetter
from django.core.exceptions import ObjectDoesNotExist
from django.urls import NoReverseMatch
class MockObject:
"""
A mock object for testing purposes.
"""
def __init__(self, **kwargs):
self._kwargs = kwargs
for key, val in kwargs.items():
setattr(self, key, val)
for key, value in kwargs.items():
setattr(self, key, value)
def __str__(self):
kwargs_str = ', '.join([
'%s=%s' % (key, value)
for key, value in sorted(self._kwargs.items())
])
return '<MockObject %s>' % kwargs_str
sorted_kwargs = ', '.join(['%s=%s' % (key, value) for key, value in sorted(self._kwargs.items())])
return f'<MockObject {sorted_kwargs}>'
class MockQueryset:
"""
A mock queryset for testing purposes.
"""
def __init__(self, iterable):
self.items = iterable
def __getitem__(self, val):
return self.items[val]
def __getitem__(self, index):
return self.items[index]
def get(self, **lookup):
for item in self.items:
if all([
attrgetter(key.replace('__', '.'))(item) == value
for key, value in lookup.items()
]):
if all(attrgetter(key.replace('__', '.'))(item) == value for key, value in lookup.items()):
return item
raise ObjectDoesNotExist()
class BadType:
"""
When used as a lookup with a `MockQueryset`, these objects
will raise a `TypeError`, as occurs in Django when making
queryset lookups with an incorrect type for the lookup value.
Raise a `TypeError` when used as a lookup value with a `MockQueryset`.
This behavior mimics Django's queryset lookups with incorrect types.
"""
def __eq__(self):
@ -47,13 +46,20 @@ class BadType:
def mock_reverse(view_name, args=None, kwargs=None, request=None, format=None):
"""
Mocked implementation of Django's reverse function for testing.
"""
args = args or []
kwargs = kwargs or {}
value = (args + list(kwargs.values()) + ['-'])[0]
first_value = (args + list(kwargs.values()) + ['-'])[0]
prefix = 'http://example.org' if request else ''
suffix = ('.' + format) if (format is not None) else ''
return '%s/%s/%s%s/' % (prefix, view_name, value, suffix)
suffix = f'.{format}' if format is not None else ''
return f'{prefix}/{view_name}/{first_value}{suffix}/'
def fail_reverse(view_name, args=None, kwargs=None, request=None, format=None):
"""
Mocked implementation of Django's reverse function to raise a NoReverseMatch for testing.
"""
raise NoReverseMatch()