Switch django default connection inside of context. It is very useful when you use readonly replica in Jupyter for analytics, but time to time you need to implement changes in master db

overwrite_default_connection.py

from contextlib import ContextDecorator
from django.db import connections

class overwrite_default_connection(ContextDecorator):
    prev_default = None
    write_connection = None
    
    def __init__(self, write_connection):
        self.write_connection = write_connection 
        
    def __enter__(self):
        self.prev_default = connections['default']
        connections['default'] = connections[self.write_connection]
        return self

    def __exit__(self, *exc):
        connections['default'] = self.prev_default
        return False
    
# with overwrite_default_connection('master_py'):
#    print('Do something with master connection')