larxius commited on
Commit
5675cc6
·
verified ·
1 Parent(s): 54d5b43

Update backend/middleware.py

Browse files
Files changed (1) hide show
  1. backend/middleware.py +6 -2
backend/middleware.py CHANGED
@@ -94,8 +94,12 @@ import ipaddress
94
 
95
 
96
 
97
- from .extensions import db, celery, socketio, limiter
98
- from .models import *
 
 
 
 
99
 
100
 
101
  # --- From security_middleware.py ---
 
94
 
95
 
96
 
97
+ try:
98
+ from .extensions import db, celery, socketio, limiter
99
+ from .models import *
100
+ except ImportError:
101
+ from extensions import db, celery, socketio, limiter
102
+ from models import *
103
 
104
 
105
  # --- From security_middleware.py ---