전체 코드 블록에 대해 db를 변경하는 방법

Question

전체 코드 블록에 대해 데이터베이스를 변경하는 방법이 있습니까? 예를 들어 : 난 그냥 상황에 따라 서로 다른 DB를 사용할 필요가 모든 시간을() 방법을 사용하여 을 사용하는 아이디어를 좋아하지 않는다

MyModel.objects.using('my_other_db_conf').all() 

:

with using_db('my_other_db_conf'): 
    MyModel.objects.all() 

의 동등 할 것이다 : \

Answer 1

관리자를 사용하고 싶습니다. 당신의 models.py에서 : 기본적 관리자가 단순히 객체

Answer 2

동작은, 일부 글로벌 값을 수정할 필요로 이름을 바꿉니다로 objects_db_one 또는 objects_db_two을 사용하려면

는 또는 IMO, 따라서 with statement은 아니다

class DB_one_ItemsManager(models.Manager): 
    def get_query_set(self): 
     return super(DB_one_ItemsManager, self).get_query_set().using("database1") 

class DB_two_ItemsManager(models.Manager): 
    def get_query_set(self): 
     return super(DB_two_ItemsManager, self).get_query_set().using("database2") 

class YourModel(models.Model):  
    #Some fields here 
    #... 

    objects_db_one=DB_one_ItemsManager() 
    objects_db_two=DB_two_ItemsManager() 

적절한 방법.

qs = User.objects.db_manager('slice_2') 
foo = qs.filter(...) 
bar = qs.filter(...) 

:

from contextlib import contextmanager 
@contextmanager 
def unsafe_modify_queryset_db(model_dbs): 
    """model_dbs => sequence of tuple (model, db for the model to use). 

    For example ((User, 'slice_2'), ...) 
    """ 
    prev_db = map(lambda x:x[0].objects._db , model_dbs) 
    for model, db in model_dbs: 
     model.objects._db = db 
    yield 
    # restore previous db 
    for x, db in zip(model_dbs, prev_db): 
     x[0].objects._db = prev_db 

# then 
with unsafe_modify_queryset_db((User, 'slice_2', ...)): 
    User.objects.filter(...) 

또한 luke14free의 코드와 같은 목표를 달성 검색어 세트 수준에서 작동하도록 db_manager을 사용할 수 물론,은, 암시 더러운 및 스레드 안전하지 않은 방법으로 수행 할 수 있습니다

Explicit가 암시 적보다 낫다는 것을 기억하십시오. 코드를 정렬하고 동일한 db를 함수 호출에 포함하는 쿼리 세트를 포함하는 것이 더 좋을 것입니다.