django_player_shard_router.py

# coding: utf-8
 
import re
 
 
class PlayerShardRouter(object):
 
  regex_app = re.compile('.*player.*')
  regex_db = re.compile('.*shard.*')
 
  def get_shard(self, playe_id):
    return 'shard'
 
  def db_for_read(self, model, **hints):
    if self.regex_app.match(model._meta.app_label):
      return self.get_shard(model.player_id)
    return None
 
  def db_for_write(self, model, **hints):
    if self.regex_app.match(model._meta.app_label):
      return self.get_shard(model.player_id)
    return None
 
  def allow_relation(self, obj1, obj2, **hints):
    if self.regex_app.match(obj1._meta.app_label) or self.regex_app.match(obj2._meta.app_label):
      return True
    return None
 
  def allow_syncdb(self, db, model):
    if regex_db.match(db):
      return regex_app.match(model._meta.app_label)
    elif regex_app.match(model._meta.app_label):
      return False
    return None