djangocms_blog/tests/test_utils/models.py

11 lines
240 B
Python
Raw Normal View History

# -*- coding: utf-8 -*-
from django.contrib.auth.models import AbstractUser
from django.db import models
class CustomUser(AbstractUser):
sites = models.ManyToManyField('sites.Site')
def get_sites(self):
return self.sites