Skip to content

Commit b20da1d

Browse files
committed
reformat with black
1 parent 5d90099 commit b20da1d

File tree

5 files changed

+62
-32
lines changed

5 files changed

+62
-32
lines changed

django_async_orm/apps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@
55

66

77
class AsyncOrmConfig(AppConfig):
8-
name = 'django_async_orm'
8+
name = "django_async_orm"
99

1010
def ready(self):
11-
logging.info('AsyncORM: patching models')
11+
logging.info("AsyncORM: patching models")
1212
for model in apps.get_models(include_auto_created=True):
1313
patch_manager(model)
1414
# TODO: patch_model(model)

django_async_orm/manager.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
2-
31
from django.db.models.manager import BaseManager
42

53
from django_async_orm.query import QuerySetAsync

django_async_orm/query.py

Lines changed: 53 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010

1111
class QuerySetAsync(QuerySet):
12-
1312
def __init__(self, model=None, query=None, using=None, hints=None):
1413
super().__init__(model, query, using, hints)
1514

@@ -20,18 +19,24 @@ async def async_create(self, **kwargs):
2019
return await sync_to_async(self.create, thread_sensitive=True)(**kwargs)
2120

2221
async def async_bulk_create(self, obs, batch_size=None, ignore_conflicts=False):
23-
return await sync_to_async(self.bulk_create, thread_sensitive=True)(obs, batch_size=batch_size,
24-
ignore_conflicts=ignore_conflicts)
22+
return await sync_to_async(self.bulk_create, thread_sensitive=True)(
23+
obs, batch_size=batch_size, ignore_conflicts=ignore_conflicts
24+
)
2525

2626
async def async_bulk_update(self, objs, fields, batch_size=None):
27-
return await sync_to_async(self.bulk_update, thread_sensitive=True)(objs=objs, fields=fields,
28-
batch_size=batch_size)
27+
return await sync_to_async(self.bulk_update, thread_sensitive=True)(
28+
objs=objs, fields=fields, batch_size=batch_size
29+
)
2930

3031
async def async_get_or_create(self, defaults=None, **kwargs):
31-
return await sync_to_async(self.get_or_create, thread_sensitive=True)(defaults=defaults, **kwargs)
32+
return await sync_to_async(self.get_or_create, thread_sensitive=True)(
33+
defaults=defaults, **kwargs
34+
)
3235

3336
async def async_update_or_create(self, defaults=None, **kwargs):
34-
return await sync_to_async(self.update_or_create, thread_sensitive=True)(defaults=defaults, **kwargs)
37+
return await sync_to_async(self.update_or_create, thread_sensitive=True)(
38+
defaults=defaults, **kwargs
39+
)
3540

3641
async def async_earliest(self, *fields):
3742
return await sync_to_async(self.earliest, thread_sensitive=True)(*fields)
@@ -48,8 +53,10 @@ async def async_none(self):
4853
async def async_last(self):
4954
return await sync_to_async(self.last, thread_sensitive=True)()
5055

51-
async def async_in_bulk(self, id_list=None, *_, field_name='pk'):
52-
return await sync_to_async(self.in_bulk, thread_sensitive=True)(id_list=id_list, *_, field_name=field_name)
56+
async def async_in_bulk(self, id_list=None, *_, field_name="pk"):
57+
return await sync_to_async(self.in_bulk, thread_sensitive=True)(
58+
id_list=id_list, *_, field_name=field_name
59+
)
5360

5461
async def async_delete(self):
5562
return await sync_to_async(self.delete, thread_sensitive=True)()
@@ -64,11 +71,14 @@ async def async_count(self):
6471
return await sync_to_async(self.count, thread_sensitive=True)()
6572

6673
async def async_explain(self, *_, format=None, **options):
67-
return await sync_to_async(self.explain, thread_sensitive=True)(*_, format=format, **options)
74+
return await sync_to_async(self.explain, thread_sensitive=True)(
75+
*_, format=format, **options
76+
)
6877

6978
async def async_raw(self, raw_query, params=None, translations=None, using=None):
70-
return await sync_to_async(self.raw, thread_sensitive=True)(raw_query, params=params, translations=translations,
71-
using=using)
79+
return await sync_to_async(self.raw, thread_sensitive=True)(
80+
raw_query, params=params, translations=translations, using=using
81+
)
7282

7383
def __aiter__(self):
7484
self._fetch_all()
@@ -78,9 +88,6 @@ def _fetch_all(self):
7888
with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
7989
future_fetch_all = executor.submit(super(QuerySetAsync, self)._fetch_all)
8090

81-
82-
83-
8491
##################################################################
8592
# PUBLIC METHODS THAT ALTER ATTRIBUTES AND RETURN A NEW QUERYSET #
8693
##################################################################
@@ -95,10 +102,14 @@ async def async_exclude(self, *args, **kwargs):
95102
return await sync_to_async(self.exclude, thread_sensitive=True)(*args, **kwargs)
96103

97104
async def async_complex_filter(self, filter_obj):
98-
return await sync_to_async(self.complex_filter, thread_sensitive=True)(filter_obj)
105+
return await sync_to_async(self.complex_filter, thread_sensitive=True)(
106+
filter_obj
107+
)
99108

100109
async def async_union(self, *other_qs, all=False):
101-
return await sync_to_async(self.union, thread_sensitive=True)(*other_qs, all=all)
110+
return await sync_to_async(self.union, thread_sensitive=True)(
111+
*other_qs, all=all
112+
)
102113

103114
async def async_intersection(self, *other_qs):
104115
return await sync_to_async(self.intersection, thread_sensitive=True)(*other_qs)
@@ -107,24 +118,38 @@ async def async_difference(self, *other_qs):
107118
return await sync_to_async(self.difference, thread_sensitive=True)(*other_qs)
108119

109120
async def async_select_for_update(self, nowait=False, skip_locked=False, of=()):
110-
return await sync_to_async(self.select_for_update, thread_sensitive=True)(nowait=nowait,
111-
skip_locked=skip_locked, of=of)
121+
return await sync_to_async(self.select_for_update, thread_sensitive=True)(
122+
nowait=nowait, skip_locked=skip_locked, of=of
123+
)
112124

113125
async def async_prefetch_related(self, *lookups):
114-
return await sync_to_async(self.prefetch_related, thread_sensitive=True)(*lookups)
126+
return await sync_to_async(self.prefetch_related, thread_sensitive=True)(
127+
*lookups
128+
)
115129

116130
async def async_annotate(self, *args, **kwargs):
117-
return await sync_to_async(self.annotate, thread_sensitive=True)(*args, **kwargs)
131+
return await sync_to_async(self.annotate, thread_sensitive=True)(
132+
*args, **kwargs
133+
)
118134

119135
async def async_order_by(self, *field_names):
120136
return await sync_to_async(self.order_by, thread_sensitive=True)(*field_names)
121137

122138
async def async_distinct(self, *field_names):
123139
return await sync_to_async(self.distinct, thread_sensitive=True)(*field_names)
124140

125-
async def async_extra(self, select=None, where=None, params=None, tables=None, order_by=None, select_params=None):
126-
return await sync_to_async(self.extra, thread_sensitive=True)(select, where, params, tables, order_by,
127-
select_params)
141+
async def async_extra(
142+
self,
143+
select=None,
144+
where=None,
145+
params=None,
146+
tables=None,
147+
order_by=None,
148+
select_params=None,
149+
):
150+
return await sync_to_async(self.extra, thread_sensitive=True)(
151+
select, where, params, tables, order_by, select_params
152+
)
128153

129154
async def async_reverse(self):
130155
return await sync_to_async(self.reverse, thread_sensitive=True)()
@@ -139,11 +164,13 @@ async def async_using(self, alias):
139164
return await sync_to_async(self.using, thread_sensitive=True)(alias)
140165

141166
async def async_resolve_expression(self, *args, **kwargs):
142-
return await sync_to_async(self.resolve_expression, thread_sensitive=True)(*args, **kwargs)
167+
return await sync_to_async(self.resolve_expression, thread_sensitive=True)(
168+
*args, **kwargs
169+
)
143170

144171
@property
145172
async def async_ordered(self):
146173
def _ordered():
147174
return super(QuerySetAsync, self).ordered
148-
return await sync_to_async(_ordered, thread_sensitive=True)()
149175

176+
return await sync_to_async(_ordered, thread_sensitive=True)()

django_async_orm/utils.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,12 @@ def mixin_async_manager_factory(model):
1212
"""
1313

1414
base_manager_cls = model.objects.__class__
15-
if not base_manager_cls.__name__.startswith('MixinAsync'):
16-
mixin_async_manager = type(f'MixinAsync{base_manager_cls.__name__}', (AsyncManager, base_manager_cls), dict())
15+
if not base_manager_cls.__name__.startswith("MixinAsync"):
16+
mixin_async_manager = type(
17+
f"MixinAsync{base_manager_cls.__name__}",
18+
(AsyncManager, base_manager_cls),
19+
dict(),
20+
)
1721
return mixin_async_manager
1822

1923

django_async_orm/wrappers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
def _sync_render(*args, **kwargs):
66
from django.shortcuts import render
7+
78
return render(*args, **kwargs)
89

910

0 commit comments

Comments
 (0)