def test_date(self):
today = datetime.date(2013, 3, 1)
date_type = DateType()
self.assertEqual(date_type("2013-03-01"), today)
self.assertEqual(date_type.to_primitive(today), "2013-03-01")
def test_date():
today = datetime.date(2013, 3, 1)
date_type = DateType()
assert date_type("2013-03-01") == today
assert date_type.to_primitive(today) == "2013-03-01"
def test_date(self):
today = datetime.date(2013, 3, 1)
date_type = DateType()
self.assertEqual(date_type("2013-03-01"), today)
self.assertEqual(date_type.to_primitive(today), "2013-03-01")
def test_date():
today = datetime.date(2013, 3, 1)
date_type = DateType()
assert date_type("2013-03-01") == today
assert date_type.to_primitive(today) == "2013-03-01"
class FormerTenure(Tenure):
start = DateType(default=None)
end = DateType(default=None)
@classmethod
def _claim_polymorphic(cls, data):
return data.get("tenure_type") == TenureType.FORMER
class Transaction(TransactionEntity):
date = DateType()
cleared = StringType()
accepted = BooleanType()
dateEnteredFromSchedule = DateType()
accountId = StringType()
payeeId = StringType()
subTransactions = ListType(ModelType(SubTransaction))
class RecurrentEntryValidationModel(BaseValidationModel):
frequency = StringType(choices=[
"yearly", "half-yearly", "quarter-yearly", "bimonthly", "monthly",
"weekly", "daily"
],
required=True)
start = DateType(formats=("%Y-%m-%d", PERIOD_DATE_FORMAT))
end = DateType(formats=("%Y-%m-%d", PERIOD_DATE_FORMAT))
class CityHallContractItem(BaseModel):
contract_id = StringType(required=True)
starts_at = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
summary = StringType()
contractor_document = StringType()
contractor_name = StringType()
value = StringType()
ends_at = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
files = ListType(StringType)
def test_date():
today = datetime.date(2013, 3, 1)
date_type = DateType()
assert date_type("2013-03-01") == today
assert date_type.to_primitive(today) == "2013-03-01"
assert date_type.to_native(today) is today
with pytest.raises(ConversionError):
date_type.to_native('foo')
class Claim(Model):
"""This model represents each claim submitted by a provider."""
splt_clm_id = StringType() # Split claim ID
clm_rndrg_prvdr_npi_num = StringType() # Unique ID of the provider
clm_rndrg_prvdr_tax_num = StringType() # Tax ID of the provider
bene_sk = StringType() # Unique ID of beneficiary
clm_ptnt_birth_dt = DateType() # Date of birth of beneficiary
clm_bene_sex_cd = StringType(
) # Sex of beneficiary. 1 for male, 2 for female
clm_from_dt = DateType() # Claim from date
clm_thru_dt = DateType() # Claim through date
dx_codes = ListType(StringType) # List of diagnosis codes
claim_lines = ListType(ModelType(
claim_line.ClaimLine)) # The associated claim lines
aggregated_procedure_codes = DictType(
BooleanType) # Map of procedure codes in all claim lines.
def __init__(self, *args, **kwargs):
"""Initialize a Claim object, calculating and storing beneficiary age as float."""
super(Claim, self).__init__(*args, **kwargs)
# Store patient age in years at date of service.
age_delta = relativedelta(self.clm_from_dt, self.clm_ptnt_birth_dt)
self.bene_age = age_delta.years + age_delta.months / 12.0 + age_delta.days / 365.0
def get_procedure_codes(self):
"""
Return the map of procedure codes present in all claim lines under this claim.
Check that codes exist. If this claim was not created by claim_reader (e.g. in a unit test)
then aggregated_procedure_codes will be null so we need to populate it.
"""
if not self.aggregated_procedure_codes:
self.aggregated_procedure_codes = {
line.clm_line_hcpcs_cd: True
for line in self.claim_lines
}
return self.aggregated_procedure_codes
def __str__(self):
"""Return a string representation of the claim."""
return 'Claim-Split Claim ID: {splt_clm_id}, npi: {npi}, claim_lines: {claim_lines}'.format(
splt_clm_id=self.splt_clm_id,
npi=self.clm_rndrg_prvdr_npi_num,
claim_lines=len(self.claim_lines or []))
class LegacyGazetteItem(BaseModel):
title = StringType(required=True)
published_on = StringType(required=False)
# important info but not available in years like 2010
date = DateType(required=False)
details = StringType(required=True)
files = ListType(StringType)
class CityHallPaymentsItem(BaseModel):
published_at = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
phase = StringType()
company_or_person = StringType(required=True)
value = StringType(required=True)
number = StringType()
document = StringType(required=True)
date = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
process_number = StringType()
summary = StringType()
group = StringType()
action = StringType()
function = StringType()
subfunction = StringType()
type_of_process = StringType()
resource = StringType()
class Flight(Model):
FL_DATE = DateType()
AIRLINE_ID = IntType()
CARRIER = StringType()
TAIL_NUM = StringType()
FL_NUM = IntType()
ORIGIN_AIRPORT_ID = IntType()
ORIGIN_AIRPORT_SEQ_ID = IntType()
ORIGIN_CITY_MARKET_ID = IntType()
ORIGIN = StringType()
ORIGIN_CITY_NAME = StringType()
ORIGIN_STATE_ABR = StringType()
ORIGIN_STATE_FIPS = IntType()
ORIGIN_STATE_NM = StringType()
ORIGIN_WAC = IntType()
DEST_AIRPORT_ID = IntType()
DEST_AIRPORT_SEQ_ID = IntType()
DEST_CITY_MARKET_ID = IntType()
DEST = StringType()
DEST_CITY_NAME = StringType()
DEST_STATE_ABR = StringType()
DEST_STATE_FIPS = IntType()
DEST_STATE_NM = StringType()
DEST_WAC = IntType()
DEP_DELAY = FloatType()
TAXI_OUT = FloatType()
WHEELS_OFF = FloatType()
WHEELS_ON = FloatType()
TAXI_IN = FloatType()
ARR_DELAY = FloatType()
AIR_TIME = FloatType()
DISTANCE = FloatType()
class GazetteItem(BaseModel):
date = DateType()
power = StringType(required=True)
year_and_edition = StringType(required=True)
events = ListType(DictType(StringType), required=True)
file_urls = ListType(StringType, required=True)
file_content = StringType()
class ProjectContribDTO(Model):
date = DateType(required=True)
mapped = IntType(required=True)
validated = IntType(required=True)
cumulative_mapped = IntType(required=False)
cumulative_validated = IntType(required=False)
total_tasks = IntType(required=False)
class TaskStats(Model):
""" DTO for tasks stats for a single day """
date = DateType(required=True)
mapped = IntType(serialized_name="mapped")
validated = IntType(serialized_name="validated")
bad_imagery = IntType(serialized_name="badImagery")
class Client(Model):
_id = StringType(default=generate_uuid())
name = StringType(required=True, min_length=1, max_length=80)
cpf = StringType(required=True, min_length=11, max_length=11)
birthdate = DateType(required=True)
amount = DecimalType(required=True, validators=[is_between])
terms = IntType(required=True, validators=[is_inside])
income = DecimalType(required=True)
class LegacyGazetteItem(BaseModel):
title = StringType(required=True)
published_on = StringType(required=False)
# important info but not available in years like 2010
date = DateType(required=False, formats=("%d/%m/%Y", "%d/%m/%y"))
details = StringType(required=True)
file_urls = ListType(URLType)
file_content = StringType()
class GazetteEventItem(BaseModel):
date = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
power = StringType(required=True)
year_and_edition = StringType(required=True)
event_title = StringType(required=True)
event_secretariat = StringType(required=True)
event_summary = StringType(required=True)
file_urls = ListType(URLType)
file_content = StringType()
class PaymentSettlementSummary(BaseModel, metaclass=ORMMeta):
"""
{
"capture_submit_time": "2016-01-21T17:15:000Z",
"captured_date": "2016-01-21"
}
"""
capture_submit_time: datetime = ArrowDTType()
captured_date: date = DateType()
class Account(DeletableEntity):
accountName = StringType()
accountType = StringType()
hidden = BooleanType()
onBudget = BooleanType()
lastReconciledDate = DateType()
lastReconciledBalance = FloatType()
lastEnteredCheckNumber = FloatType()
sortableIndex = IntType()
note = StringType()
class DocumentData(Model):
address_history: Optional[List[DatedAddress]] = ListType(
ModelType(DatedAddress), default=None)
expiry = DateType(default=None)
external_ref = StringType(default=None)
external_service = StringType(default=None)
issued = DateType(default=None)
issuer = StringType(default=None)
issuing_country = StringType(default=None)
mrz1 = StringType(default=None)
mrz2 = StringType(default=None)
mrz3 = StringType(default=None)
number = StringType(default=None)
personal_details: Optional[PersonalDetails] = ModelType(PersonalDetails,
default=None)
result: Optional[DocumentResult] = ModelType(DocumentResult, default=None)
class Options:
export_level = NOT_NONE
class ClaimLine(Model):
"""
This model contains the claim-line specific information.
Each claim will have multiple claim lines. Claim lines contain all of the encounter
or procedure codes that took place in a visit.
"""
clm_line_num = IntType() # Claim line number
clm_line_hcpcs_cd = StringType() # HCPCS or C4 CPT code at the claim line level
mdfr_cds = ListType(StringType) # Come from 5 columns, some of which will be null
clm_pos_code = StringType() # Place of service
clm_line_from_dt = DateType() # Beginning date of service for the line item
clm_line_thru_dt = DateType() # Final date of service for the line item
def __str__(self):
"""Return a string representation of the claim."""
# TODO - Add more relevant info to __repr__.
return 'ClaimLine - line_number: {clm_line_num}'.format(clm_line_num=self.clm_line_num)
class Company(Model): name = StringType(required=True) full_name = StringType(required=True) domain = StringType(required=True) founded = DateType() headquarters = ModelType(Headquarters) long_description = StringType(required=True) short_description = StringType(required=True) technology_labels = ListType(StringType()) industry_labels = ListType(StringType()) num_employees = IntType()
class Eventos(BaseModel):
nome = StringType(max_length=100, required=True)
data = DateType()
local = StringType(max_length=100)
endereco = StringType(max_length=100)
site = StringType(max_length=100)
class Meta:
list_allowed_methods = [
'get',
]
class TCMBADocumentItem(Model):
crawled_at = DateTimeType(required=True)
category = StringType()
filename = StringType(required=True)
original_filename = StringType(required=True)
filepath = StringType(required=True)
inserted_by = StringType()
inserted_at = DateType(formats=("%d/%m/%Y", "%d/%m/%y"))
unit = StringType(required=True)
month = StringType()
year = StringType()
period = StringType()
class SignupInputs(Model):
email = LowerCaseEmailType(required=True)
password = StringType(min_length=6, max_length=12, required=True)
full_name = StringType(required=True)
phone_no = IntType(min_value=1000000000, max_value=9999999999,
required=True)
college_id = IntType(min_value=0)
referral_code_used = StringType(required=False)
graduation_date = DateType(required=False)
degree = StringType(required=False)
branch = StringType(required=False)
captcha_token = ReCaptchaValidator(required=True)
class UserContribution(Model):
""" User contribution for a project """
username = StringType()
mapping_level = StringType(serialized_name="mappingLevel")
picture_url = StringType(serialized_name="pictureUrl")
mapped = IntType()
validated = IntType()
total = IntType()
mapped_tasks = ListType(IntType, serialized_name="mappedTasks")
validated_tasks = ListType(IntType, serialized_name="validatedTasks")
name = StringType()
date_registered = DateType(serialized_name="dateRegistered")
class CourtOrderESAJModel(Model):
number = StringType(required=True)
decision = StringType(required=True)
decision_date = DateType(required=True)
status = StringType()
source_numbers = StringType()
reporter = StringType()
category = StringType()
subject = StringType()
petitioner = StringType()
petitioner_attorneys = StringType()
requested = StringType()
requested_attorneys = StringType()
appeals = StringType()
class PostItem(Model):
visit_ts = DateTimeType(required=True)
date = DateType(required=True)
link = URLType(required=True)
parent_link = URLType(required=True)
campus = StringType(required=True)
sections = StringType(required=True)
title = StringType(required=True)
description = StringType()
text = StringType(required=True)
tags = ListType(StringType)
people = ListType(StringType)
branches = ListType(StringType)
def test_date():
today = datetime.date(2013, 3, 1)
date_type = DateType()
assert date_type("2013-03-01") == today
assert date_type.to_primitive(today) == "2013-03-01"
assert date_type.to_native(today) is today
with pytest.raises(ConversionError):
date_type.to_native('foo')
def field_class_to_schematics_field(field: peewee.Field) -> BaseType:
if isinstance(field, peewee.ForeignKeyField):
field = field.rel_field
kwargs = {}
# 检查是否 require
if not ((field.default is not None) or field.null or field.sequence
or isinstance(field, peewee.AutoField)):
kwargs['required'] = True
if field.help_text:
kwargs['metadata'] = {'description': field.help_text}
if isinstance(field, peewee.IntegerField):
return IntType(**kwargs)
elif isinstance(field, peewee.FloatField):
return FloatType(**kwargs)
elif isinstance(field,
(PG_JSONField, PG_BinaryJSONField, SQLITE_JSONField)):
# 注意 SQLITE_JSONField 是一个 _StringField 所以要提前
return JSONType(**kwargs)
# HStore 貌似才应该对应 dict，json可以对应任意类型
# return JSONDictType(StringType, **kwargs)
elif isinstance(field, peewee.DateTimeField):
return DateTimeType(**kwargs)
elif isinstance(field, peewee.DateField):
return DateType(**kwargs)
elif isinstance(field, peewee._StringField):
return StringType(**kwargs)
elif isinstance(field, peewee.BooleanField):
return BooleanType(**kwargs)
elif isinstance(field, peewee.BlobField):
return BlobType(**kwargs)
elif isinstance(field, PG_ArrayField):
field: PG_ArrayField
return JSONListType(
field_class_to_schematics_field(field._ArrayField__field),
**kwargs)
class CompanyMetadata(Model):
name = StringType(default=None)
number = StringType(default=None)
lei = StringType(default=None)
isin = StringType(default=None)
lei = StringType(default=None)
company_type = StringType(default=None)
# structured_company_type = ModelType(StructuredCompanyType)
industry_classifications = ListType(ModelType(IndustryClassification),
default=None)
# tax_ids = ListType(ModelType(TaxId), default=list, required=True)
country_of_incorporation = StringType(default=None)
incorporation_date = DateType(default=None)
contact_information = ModelType(ContactDetails, default=None)
addresses = ListType(ModelType(CompanyAddress), default=None)
is_active = BooleanType(default=None)
is_active_details = StringType(default=None)
trade_description = StringType(default=None)
description = StringType(default=None)
