def test_unique():
field = AutoField(name='testname', unique=True)
field.model = Mock(**{'_meta.model_name': 'testmodel'})
field.attname = 'testattname'
field = fields.copy_other(field)
assert field.name == 'testattname'
assert not field.unique
assert field.db_index
def test_primary():
field = AutoField(name='testname', primary_key=True)
field.model = Mock(**{'_meta.model_name': 'testmodel'})
field.attname = 'testattname'
field = fields.copy_other(field)
assert field.name == 'testmodel'
assert not field.unique
assert not field.primary_key
assert field.db_index
assert field.blank
assert field.null
class Client(DBModel):
id = AutoField(primary_key=True)
first_name = CharField(_('First name'), help_text=_('First name'), max_length=100)
last_name = CharField(_('Last name'), help_text=_('Last name'), max_length=100)
id_no = CharField(_('Id #'), help_text=_('ID'), max_length=50, unique=True, validators=[validate_id_no])
address = CharField(_('Address'), help_text=_('Postal address'), blank=True, null=True, max_length=100)
zip_code = PositiveSmallIntegerField(_('Zip code'), help_text=_('Address zip code'), blank=True, null=True)
email = EmailField(blank=True, null=True, unique=False)
file_no = CharField(_('File #'), help_text=_('Internal file #'), max_length=50, blank=True, null=True, default=None)
notes = TextField(_('Notes'), help_text=_('Notes'), blank=True, null=True)
class Meta:
verbose_name = _('client')
verbose_name_plural = _('clients')
def __str__(self):
return '<Client: %s %s, id: %s, file_no: %s>' % (self.first_name, self.last_name,
self.id_no, self.file_no)
def __repr__(self):
return self.__str__()
def __unicode__(self):
return '%s, %s (%s)' % (self.last_name, self.first_name, self.file_no)
class UserInfo(Model):
id = AutoField(primary_key=True, verbose_name="用户ID")
name = CharField(max_length=20, verbose_name="用户名称", null=True)
sex = CharField(max_length=20, verbose_name="性别", null=True)
age = IntegerField(verbose_name="年龄", null=True)
height = CharField(max_length=20, verbose_name="身高/cm", null=True)
weight = CharField(max_length=20, verbose_name="体重", null=True)
address = CharField(max_length=50, verbose_name="住址", null=True)
id_number = CharField(max_length=30, verbose_name="身份证号", null=True)
tel = CharField(max_length=20, verbose_name="电话号码", null=True)
edu_background = CharField(max_length=20, verbose_name="文化程度", null=True)
is_instructor = BooleanField(verbose_name="是否指导员", null=False)
instructor_class = CharField(max_length=20,
verbose_name="指导员等级",
null=True)
instructor_type = CharField(max_length=40, verbose_name="指导类型", null=True)
ver_status = CharField(max_length=20, verbose_name="审核状态", null=True)
province = CharField(max_length=20, verbose_name="省级区划", null=True)
city = CharField(max_length=20, verbose_name="市级区划", null=True)
village = CharField(max_length=20, verbose_name="乡镇", null=True)
certificate_num = CharField(max_length=50, verbose_name="证书编号", null=True)
certificate_date = CharField(max_length=20, verbose_name="发证日期", null=True)
gym = CharField(max_length=50, verbose_name="健身地点", null=True)
assessment = CharField(max_length=30, verbose_name="评定成绩", null=True)
upload_time = CharField(max_length=30, verbose_name="上传日期", null=True)
annual_survey_date = DateField(max_length=30,
verbose_name="年检日期",
null=True)
is_active = BooleanField(null=False)
class Meta:
db_table = "userinfo"
class Operation(Model):
"""Operation model."""
class Meta:
abstract = True
PENDING = 0
ASSIGNED = 1
SENT = 2
STATUS_CHOICES = (
(PENDING, "pending"),
(ASSIGNED, "assigned"),
(SENT, "sent"),
)
id = AutoField(primary_key=True)
creation_datetime = DateTimeField(auto_now_add=True)
last_modification_datetime = DateTimeField(auto_now=True)
amount = IntegerField()
status = IntegerField(choices=STATUS_CHOICES, default=PENDING)
@property
@abstractmethod
def product(self) -> Product:
"""Product attribute.
:return: A product object.
"""
pass
class Role_Module(Model):
id = AutoField("Id", primary_key=True)
module_id = ForeignKey(Module, on_delete=Module)
role_id = ForeignKey(Role, on_delete=Role)
class Meta:
db_table = "Role_Module"
class CrawlSyncTask(Model):
class Meta:
db_table = 'crawl_sync_cfg'
id = AutoField(auto_created=True, primary_key=True)
task_name = CharField(max_length=255, null=True, blank=True)
task_desc = CharField(max_length=255, null=True, blank=True)
execute_host = CharField(max_length=255, null=True, blank=True)
source_cfg = CharField(max_length=1024, null=True, blank=True)
target_cfg = CharField(max_length=255, null=True, blank=True)
effect_date = DateTimeField(null=True, blank=True)
schedule_cycle = IntegerField(blank=True)
schedule_date = CharField(max_length=255, null=True, blank=True)
job_id = CharField(max_length=255, null=True, blank=True)
is_deleted = IntegerField(default=0, blank=True)
created_at = DateTimeField(auto_now_add=True, blank=True, null=True)
modified_at = DateTimeField(auto_now=True, blank=True, null=True)
creator_id = IntegerField(null=True, blank=True)
def to_dict(self):
opts = self._meta
data = {}
for f in opts.concrete_fields:
value = f.value_from_object(self)
if isinstance(value, datetime):
value = value.strftime('%Y-%m-%d %H:%M:%S')
data[f.name] = value
return data
class InjectResponse(Model): ijctrespid = AutoField(primary_key = True) compid = PositiveIntegerField() teamid = PositiveIntegerField() ijctid = PositiveIntegerField() datetime = DateTimeField(default = timezone.now()) textentry = TextField(max_length = 1000)
class TinhThanh(Model):
id = AutoField("Id", primary_key=True)
ten = CharField("Ten", max_length=265)
trangthai = BooleanField("TrangThai")
class Meta:
db_table = 'TinhThanh'
class Movie(Model):
id = AutoField(primary_key=True)
name = CharField(max_length=255, unique=True)
alias = CharField(max_length=255, null=True, blank=True)
cover = CharField(max_length=255, null=True, blank=True)
categories = ArrayField(CharField(max_length=255, null=True, blank=True),
null=True,
blank=True)
regions = ArrayField(CharField(max_length=255, null=True, blank=True),
null=True,
blank=True)
actors = JSONField(null=True, blank=True)
directors = JSONField(null=True, blank=True)
score = FloatField(null=True, blank=True)
rank = IntegerField(null=True, blank=True)
minute = IntegerField(null=True, blank=True)
drama = TextField(null=True, blank=True)
photos = ArrayField(CharField(max_length=500, null=True, blank=True),
null=True,
blank=True)
published_at = DateField(null=True, blank=True)
updated_at = DateTimeField(auto_now=True, null=True, blank=True)
class Meta:
app_label = 'app'
class BaseModel(Model):
class Meta:
abstract = True
id = AutoField(primary_key=True)
created_at = DateTimeField(default=timezone.now)
def _get_explicit_field_names(self):
"""
:rtype: list
"""
return [
field.name for field in self._meta.get_fields() if
isinstance(field, Field) and not isinstance(field, RelatedField)
]
def to_dict(self):
"""
:rtype: dict
"""
fields = self._get_explicit_field_names()
return {
field: getattr(self, field)
for field in fields if getattr(self, field)
}
def __repr__(self):
model = str(self.__class__.__name__)
columns = ', '.join('{}="{}"'.format(field, value)
for field, value in self.to_dict().items())
return '{}({})'.format(model, columns)
class Document(Model):
docid = AutoField(primary_key=True)
inject = ForeignKey(Inject, null=True, blank=True, unique=False)
injectresponse = ForeignKey(InjectResponse,
null=True,
blank=True,
unique=False)
incidentresponse = ForeignKey(IncidentResponse,
null=True,
blank=True,
unique=False)
servicemodule = ForeignKey(ServiceModule,
null=True,
blank=True,
unique=True)
content_type = CharField(max_length=64, null=True)
filehash = CharField(max_length=32)
filepath = CharField(max_length=256)
filename = CharField(max_length=64)
urlencfilename = CharField(max_length=128)
def get_cleaned_content_type(self):
if not self.content_type:
return 'application/force-download'
else:
return self.content_type
class CrawlScriptProgress(Model):
"""
任务脚本执行数据
"""
class Meta:
db_table = 'crawl_script_progress'
id = AutoField(auto_created=True, primary_key=True)
batch = CharField(max_length=255, null=True, blank=True, unique=True)
task_name = CharField(max_length=255, null=True, blank=True)
script_name = CharField(max_length=255, null=True, blank=True)
status = IntegerField(default=0)
start_time = CharField(max_length=255, null=True, blank=True)
end_time = CharField(max_length=255, null=True, blank=True)
run_time = IntegerField(default=0)
get_cnt = IntegerField(default=0)
request_cnt = IntegerField(default=0)
oss_url = CharField(max_length=255, null=True, blank=True)
msg = CharField(max_length=255, null=True, blank=True)
node = CharField(max_length=255, null=True, blank=True)
arg = CharField(max_length=255, null=True, blank=True)
arg_key = CharField(max_length=255, null=True, blank=True)
is_deleted = IntegerField(default=0)
created_at = DateTimeField(auto_now_add=True, blank=True, null=True)
updated_at = DateTimeField(auto_now=True, blank=True, null=True)
class CrawlTask(Model):
"""
任务表
"""
class Meta:
db_table = 'crawl_task'
id = AutoField(auto_created=True, primary_key=True)
task_name = CharField(max_length=255, null=True, blank=True)
project_name = CharField(max_length=255, null=True, blank=True)
platform_name = CharField(max_length=255, null=True, blank=True)
description = CharField(max_length=255, null=True, blank=True)
project_id = IntegerField(default=0)
platform_id = CharField(max_length=255, null=True, blank=True)
task_type = IntegerField(default=0)
is_deploy = IntegerField(default=0)
is_deleted = IntegerField(default=0)
spider_concurrency = CharField(max_length=2048, null=True, blank=True)
node_ids = CharField(max_length=255, null=True, blank=True)
status = IntegerField(default=0)
creator_id = IntegerField(null=True, blank=True)
created_at = DateTimeField(auto_now_add=True,
blank=True,
null=True,
editable=True)
updated_at = DateTimeField(auto_now=True, blank=True, null=True)
class Invoice(DBModel):
id = AutoField(primary_key=True)
number = CharField(_('Invoice #'), help_text=_('Invoice #'), unique=True, null=False, max_length=70)
date = DateField(_('Date'), help_text=_('Date'), null=False, default=date.today())
client = ForeignKey('Client', verbose_name=_('client'))
bike = ManyToManyField('Bike', verbose_name=_('bike'))
class Meta:
verbose_name = _('invoice')
verbose_name_plural = _('invoices')
#def bikes(self):
# return self.bike_set.all()
def __str__(self):
return '<Invoice: %s, date: %s>' % (self.number, self.date)
def __repr__(self):
return self.__str__()
def __unicode__(self):
ret = '%s' % (self.number)
if self.client is not None:
ret += '%s' % self.client
return ret
class Post(Model):
_id = AutoField(
db_column='_id',
name='_id',
primary_key=True,
)
titulo = CharField(db_column='titulo',
name='titulo',
null=True,
max_length=150)
tags = PgArrayField(CharField(max_length=15), null=True, blank=True)
corpo = TextField(
db_column='corpo',
name='corpo',
null=True,
)
criado = DateTimeField(db_column='criado',
name='criado',
null=False,
auto_now_add=True)
atualizado = DateTimeField(db_column='atualizado',
name='atualizado',
null=False,
auto_now_add=True)
def __str__(self):
return '{} - {}'.format(self._id, self.titulo)
def get_absolute_url(self):
return reverse('posts:detail', kwargs={'pk': self.pk})
class Meta:
db_table = 'tb_post'
verbose_name_plural = 'Post' # Tirar o 's' do plural na exibição
class TweetHashtagBase(Model):
_id = AutoField(primary_key=True, db_column='id')
text = CharField(max_length=255)
indices = CommaSeparatedIntegerField(max_length=255)
class Meta:
abstract = True
class LuaChon_SanPham(Model):
id = AutoField("Id", primary_key=True)
sanpham_id = ForeignKey(SanPham, on_delete=SanPham)
ten = CharField("Ten", max_length=265)
class Meta:
db_table = 'LuaChon_SanPham'
class Service(Model):
servid = AutoField(primary_key = True)
compid = PositiveIntegerField()
servicemodule = ForeignKey(ServiceModule, unique = False)
name = CharField(max_length = 30)
description = CharField(max_length = 200)
datetime_start = DateTimeField()
datetime_finish = DateTimeField()
points = PositiveIntegerField()
connectip = BooleanField(default = True)
connect_display = CharField(max_length = 15)
networkloc = CharField(max_length = 15)
defaultport = PositiveIntegerField()
# Service object now has the ability to score itself
def score(self, team_obj):
instance = self.load_plugin()
score_obj = instance.score(team_obj)
score_obj.datetime = timezone.now()
return score_obj
def load_plugin(self):
module_name = Document.objects.get(servicemodule = self.servicemodule).filename.split(".")[0]
module = __import__(settings.CONTENT_PLUGGINS_PATH.replace('/','.')[1:] + module_name, fromlist=[module_name])
return getattr(module, module_name)(self)
class User(AbstractUser):
# First Name and Last Name do not cover name patterns
# around the globe.
id = AutoField(primary_key=True)
first_name = CharField(_("First name of User"), blank=True, max_length=255)
last_name = CharField(_("Last name of User"), blank=True, max_length=255)
address = CharField(blank=True, max_length=255)
# password = CharField(max_length=50)
contact = PhoneNumberField(
max_length=13,
help_text="Enter phone number with country code",
null=True,
blank=True)
#phone_number = forms.RegexField(regex=r'^\+?1?\d{9,15}$',
# error_message = ("Phone number must be entered in the format: '+999999999'. Up to 15 digits allowed."))
email = EmailField(blank=True, null=True, verbose_name="Email")
date_of_birth = DateField(blank=True, null=True)
branch = ForeignKey(Branch, on_delete=CASCADE, null=True)
class Meta:
default_permissions = ()
permissions = (
("add_staff", "Can add staff"),
("accept_request", "Can accept leave request"),
("make_request", "can make leave request"),
("set_roles_of_staff", "can set roles of staff"),
("view_attendance", "can veiw attendance"),
)
def get_absolute_url(self):
return reverse("users:detail", kwargs={"username": self.username})
class Appointment(models.Model):
doctorId = ForeignKey(Doctor, on_delete=models.CASCADE)
hospitalId = ForeignKey(Hospital, on_delete=models.CASCADE)
patientId = ForeignKey(Patient, on_delete=models.CASCADE)
start = DateTimeField()
end = DateTimeField()
id = AutoField(primary_key=True)
class Representation(Model):
""" Representation entity which can be image, video, etc. """
TYPE_IMAGE_CODE = 1
TYPE_VIDEO_CODE = 2
TYPES = [(TYPE_IMAGE_CODE, 'Image'), (TYPE_VIDEO_CODE, 'Video')]
representation_id = AutoField(primary_key=True)
representation_type_code = IntegerField(null=True,
blank=False,
choices=TYPES)
article_representation = GenericRelation(
'ArticleRepresentation',
object_id_field='representation_id',
content_type_field='representation_content_type'
) #Bi-directional mapping
class Meta:
db_table = 'green_planet_representation'
def is_image(self):
""" Method returns True, if this instance is an image type """
return self.TYPE_IMAGE_CODE == self.representation_type_code
def is_video(self):
""" Method returns True, if this instance is an video type """
return self.TYPE_VIDEO_CODE == self.representation_type_code
class Module_HoatDong(Model):
id = AutoField("Id", primary_key=True)
module_id = ForeignKey(Module, on_delete=Module)
hoatdong_id = ForeignKey(HoatDong, on_delete=HoatDong)
class Meta:
db_table = "Module_HoatDong"
class ArticleKeyword(Model):
""" Article Keyword entity """
article_keyword_id = AutoField(primary_key=True)
text = CharField(max_length=255, blank=False, null=False)
class Meta:
db_table = "green_planet_article_keyword"
class Resume(Model):
id = AutoField(primary_key=True)
title = CharField(max_length=255, null=False)
author = ForeignKey(User, related_name='user')
raw = FileField()
createdAt = DateTimeField(auto_now_add=True)
class User(Model):
id = AutoField("Id", primary_key=True)
taikhoan_id = IntegerField("TaiKhoan_Id")
user_trangthai_id= ForeignKey(User_TrangThai,on_delete=User_TrangThai)
class Meta:
db_table = 'User'
class Migration(BaseMigration):
"""Represents initial migration."""
initial: bool = True
dependencies: List[str] = []
operations: List[CreateModel] = [
CreateModel(
name="Quote",
fields=[
(
"id",
AutoField(
auto_created=True,
primary_key=True,
serialize=False,
verbose_name="ID",
),
),
("quote", TextField()),
("author", CharField(max_length=100)),
("source", URLField(blank=True, null=True)),
["cover", URLField(blank=True, null=True)],
("added", DateTimeField(auto_now_add=True)),
("edited", DateTimeField(auto_now=True)),
],
options={"ordering": ["-added"]},
),
]
def _prepare(self, model):
if self.order_with_respect_to:
# The app registry will not be ready at this point, so we cannot
# use get_field().
query = self.order_with_respect_to
try:
self.order_with_respect_to = next(
f for f in self._get_fields(reverse=False)
if f.name == query or f.attname == query
)
except StopIteration:
raise FieldDoesNotExist("%s has no field named '%s'" % (self.object_name, query))
self.ordering = ('_order',)
if not any(isinstance(field, OrderWrt) for field in model._meta.local_fields):
model.add_to_class('_order', OrderWrt())
else:
self.order_with_respect_to = None
if self.pk is None:
if self.parents:
# Promote the first parent link in lieu of adding yet another
# field.
field = next(iter(self.parents.values()))
# Look for a local field with the same name as the
# first parent link. If a local field has already been
# created, use it instead of promoting the parent
already_created = [fld for fld in self.local_fields if fld.name == field.name]
if already_created:
field = already_created[0]
field.primary_key = True
self.setup_pk(field)
else:
auto = AutoField(verbose_name='ID', primary_key=True, auto_created=True)
model.add_to_class('id', auto)
class Person(Model):
'''
Person Model
Namespace: ns_hr
Table: tb_person
'''
id_ = AutoField(
db_column='id_',
name='id',
primary_key=True,
)
name = TextField(
db_column='name',
name='name',
)
surname = TextField(
db_column='surname',
name='surname',
)
def __str__(self):
return f'{self.name} {self.surname}'
class Meta:
db_table = 'ns_hr"."tb_person' # 'schema"."object'
verbose_name_plural = 'Person'
class Model(DBModel):
id = AutoField(primary_key=True)
name = CharField(_('name'), help_text=_('Model name'), max_length=50, unique=True)
cc = PositiveIntegerField(_('Cubic centimeters'), help_text=_('Cubic centimeters'), blank=True)
year = PositiveIntegerField(_('year'), help_text=_('Model year'), blank=True)
brand = ForeignKey('Brand', verbose_name=_('brand'))
class Meta:
verbose_name = _('model')
verbose_name_plural = _('models')
def __str__(self):
return '<Model: %s, brand: %s, cc: %d, year: %d>' % (self.name, self.brand.name,
self.cc, self.year)
def __repr__(self):
return self.__str__()
def __unicode__(self):
ret = self.name
if self.cc is not None and self.cc > 0:
ret += ' - %d cc' % self.cc
if self.year is not None and self.year > 0:
ret += ' - %d' % self.year
return ret
