Skip to content

Instantly share code, notes, and snippets.

@ijharulislam
Created October 7, 2017 04:32
Show Gist options
  • Save ijharulislam/547c99509907893af120156cc9b333b0 to your computer and use it in GitHub Desktop.
Save ijharulislam/547c99509907893af120156cc9b333b0 to your computer and use it in GitHub Desktop.
from account.models import Store, UserAccount
from django.utils.deprecation import MiddlewareMixin
class StoreSessionMiddleware(MiddlewareMixin):
def process_request(self, request):
if "store_id" in request.POST:
print "setting session store id"
req_store = request.POST.get("store_id")
selected_store = Store.objects.get(id=req_store)
request.session['store_id'] = selected_store.id
request.session['store_time_zone'] = selected_store.time_zone
request.session['token'] = selected_store.phone
elif "store_id" not in request.session and request.user.is_authenticated():
print "setting first store id as default"
user_account = UserAccount.objects.get(user=request.user)
selected_store = Store.objects.filter(account=user_account.account).first()
request.session['store_id'] = selected_store.id
request.session['store_time_zone'] = selected_store.time_zone
request.session['token'] = selected_store.phone
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment