from django.db import models
from django.conf import settings
class News(models.Model):
    news_id = models.AutoField(primary_key=True)
    source = models.CharField(max_length=400)
    text = models.TextField()
    image = models.FileField(upload_to=settings.UPLOADS_ROOT+"/news/")
    date = models.DateTimeField(auto_now=True)

SPR_MODELS=['pidpr','dov_post']

class reestrRouter(object):
    """A router to control all database operations"""

    def db_for_read(self, model, **hints):
        if model._meta.object_name in SPR_MODELS:
            return 'spr_db'
        else:
            return 'default'
        return None

    def db_for_write(self, model, **hints):
        return 'default'

    def allow_relation(self, obj1, obj2, **hints):
        "Allow any relation"
        return True

    def allow_syncdb(self, db, model):
        print model._meta.app_label, model._meta.object_name 
        if db == 'spr_db':
            return model._meta.object_name in SPR_MODELS
        else:
            return model._meta.object_name not in SPR_MODELS
        return None