270 lines
6.1 KiB
Python
270 lines
6.1 KiB
Python
from paramecio2.libraries.db.webmodel import PhangoField
|
|
from paramecio2.libraries.db import coreforms
|
|
from paramecio2.libraries.i18n import I18n
|
|
from bs4 import BeautifulSoup
|
|
|
|
class IntegerField(PhangoField):
|
|
|
|
"""Class that figure an integer sql type field.
|
|
|
|
Args:
|
|
name (str): The name of new field
|
|
size (int): The size of the new field in database. By default 11.
|
|
required (bool): Boolean for define if
|
|
|
|
"""
|
|
|
|
def __init__(self, name, size=11, required=False):
|
|
super(IntegerField, self).__init__(name, size, required)
|
|
self.default_value=0
|
|
|
|
def check(self, value):
|
|
|
|
"""Method for check if value is integer
|
|
|
|
Args:
|
|
value (int): The value to check
|
|
|
|
"""
|
|
|
|
self.error=False
|
|
self.txt_error=''
|
|
|
|
try:
|
|
|
|
value=str(int(value))
|
|
|
|
if value=="0" and self.required==True:
|
|
self.txt_error="The value is zero"
|
|
self.error=True
|
|
except:
|
|
|
|
value="0"
|
|
self.txt_error="The value is zero"
|
|
self.error=True
|
|
|
|
return value
|
|
|
|
def get_type_sql(self):
|
|
|
|
"""Method for return the sql code for this type
|
|
|
|
"""
|
|
|
|
return 'INT('+str(self.size)+') NOT NULL DEFAULT "0"'
|
|
|
|
class BigIntegerField(IntegerField):
|
|
|
|
"""Class that figure an big integer sql type field.
|
|
|
|
Only change the sql type with respect to IntegerField
|
|
|
|
"""
|
|
|
|
def get_type_sql(self):
|
|
|
|
"""Method for return the sql code for this type
|
|
|
|
"""
|
|
|
|
return 'BIGINT('+str(self.size)+') NOT NULL DEFAULT "0"'
|
|
|
|
|
|
class FloatField(PhangoField):
|
|
|
|
"""Class that figure an float sql type field.
|
|
|
|
Args:
|
|
name (str): The name of new field
|
|
size (int): The size of the new field in database. By default 11.
|
|
required (bool): Boolean for define if
|
|
|
|
"""
|
|
|
|
def __init__(self, name, size=11, required=False):
|
|
super(FloatField, self).__init__(name, size, required)
|
|
|
|
self.error_default="The value is zero"
|
|
self.default_value=0
|
|
|
|
def check(self, value):
|
|
|
|
"""Method for check if value is integer
|
|
|
|
Args:
|
|
value (float): The value to check
|
|
|
|
"""
|
|
|
|
self.error=False
|
|
self.txt_error=''
|
|
|
|
try:
|
|
|
|
value=str(value)
|
|
|
|
if value.find(',')!=-1:
|
|
value=value.replace(',', '.')
|
|
|
|
value=str(float(value))
|
|
|
|
if value==0 and self.required==True:
|
|
self.txt_error=self.error_default
|
|
self.error=True
|
|
except:
|
|
|
|
value="0"
|
|
self.txt_error=self.error_default
|
|
self.error=True
|
|
|
|
return value
|
|
|
|
def get_type_sql(self):
|
|
|
|
return 'FLOAT NOT NULL DEFAULT "0"'
|
|
|
|
class DecimalField(FloatField):
|
|
|
|
def get_type_sql(self):
|
|
|
|
return 'DECIMAL(20, 2) NOT NULL DEFAULT "0"'
|
|
|
|
class DoubleField(FloatField):
|
|
|
|
def get_type_sql(self):
|
|
|
|
return 'DOUBLE NOT NULL DEFAULT "0"'
|
|
|
|
class CharField(PhangoField):
|
|
|
|
pass
|
|
|
|
class TextField(PhangoField):
|
|
|
|
def __init__(self, name, required=False):
|
|
super().__init__(name, 11, required)
|
|
|
|
self.set_default='NOT NULL'
|
|
|
|
def get_type_sql(self):
|
|
|
|
"""Method for return the sql code for this type
|
|
|
|
"""
|
|
|
|
return 'TEXT '+self.set_default
|
|
|
|
class HTMLField(TextField):
|
|
|
|
def __init__(self, name, required=False):
|
|
super().__init__(name, required)
|
|
self.trusted_tags=[]
|
|
|
|
def check(self, value):
|
|
|
|
soup=BeautifulSoup(value, features='html.parser')
|
|
|
|
for tag in soup.findAll(True):
|
|
if tag.name not in self.trusted_tags:
|
|
tag.hidden=True
|
|
|
|
return soup.renderContents().decode('utf-8')
|
|
|
|
|
|
class ForeignKeyField(IntegerField):
|
|
|
|
def __init__(self, name, related_table, size=11, required=False, identifier_field='id', named_field="id", select_fields=[]):
|
|
|
|
super(ForeignKeyField, self).__init__(name, size, required)
|
|
|
|
self.table_id=related_table.name_field_id
|
|
|
|
self.table_name=related_table.name
|
|
|
|
self.related_model=related_table
|
|
|
|
self.identifier_field=identifier_field
|
|
|
|
self.named_field=named_field
|
|
|
|
self.select_fields=select_fields
|
|
|
|
self.foreignkey=True
|
|
|
|
self.change_form(coreforms.SelectModelForm, [related_table, self.named_field, self.identifier_field])
|
|
|
|
self.default_value=None
|
|
|
|
def check(self, value):
|
|
|
|
value=super().check(value)
|
|
|
|
if value=='0' or value==0:
|
|
value='NULL'
|
|
|
|
return value
|
|
|
|
def get_type_sql(self):
|
|
|
|
"""Method for return the sql code for this type
|
|
|
|
"""
|
|
|
|
return 'INT NULL'
|
|
|
|
|
|
class BooleanField(IntegerField):
|
|
|
|
def __init__(self, name, size=1):
|
|
|
|
required=False
|
|
|
|
self.yes_text=I18n.lang('common', 'yes', 'Yes')
|
|
self.no_text=I18n.lang('common', 'no', 'No')
|
|
|
|
super(IntegerField, self).__init__(name, size, required)
|
|
|
|
self.default_error="Need 0 or 1 value"
|
|
self.default_value=0
|
|
|
|
def check(self, value):
|
|
|
|
self.error=False
|
|
self.txt_error=''
|
|
|
|
try:
|
|
|
|
value=int(value)
|
|
|
|
if value<0 or value>1:
|
|
self.txt_error=self.default_error
|
|
self.error=True
|
|
|
|
except:
|
|
|
|
self.error=True
|
|
self.txt_error=self.default_error
|
|
value=0
|
|
|
|
value=str(value)
|
|
|
|
return value
|
|
|
|
def get_type_sql(self):
|
|
|
|
"""Method for return the sql code for this type
|
|
|
|
"""
|
|
|
|
return 'BOOLEAN NOT NULL DEFAULT "0"'
|
|
|
|
def show_formatted(self, value):
|
|
|
|
value=int(value)
|
|
|
|
if value==0:
|
|
value=self.no_text
|
|
else:
|
|
value=self.yes_text
|
|
|
|
return str(value)
|