Skip to content

Commit ab26f19

Browse files
committed
Fixed issue with initializing the AMQP extension
1 parent a5f6768 commit ab26f19

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sage_utils/amqp/extension.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77

88
class AmqpExtension(BaseExtension):
9+
extension_name = 'amqp'
910
app_attribute = 'amqp'
1011
workers = []
1112
active_tasks = []
@@ -31,6 +32,7 @@ async def connect(self):
3132
async def init(self, loop):
3233
if not hasattr(self.app, 'extensions'):
3334
setattr(self.app, 'extensions', {})
35+
setattr(self.app, self.app_attribute, self)
3436
self.app.extensions[self.extension_name] = self
3537

3638
for worker in self.workers:
@@ -47,4 +49,4 @@ async def deinit(self, loop):
4749

4850
setattr(self.app, self.app_attribute, None)
4951
extensions = getattr(self.app, 'extensions', {})
50-
extensions.pop(self.extension_name, None)
52+
extensions.pop(self.extension_name, None)

0 commit comments

Comments
 (0)