class ESRelatedModel(BaseESModel): """Elasticsearch representation of SimpleModel model.""" id = Keyword() simpleton = fields.nested_id_name_field() MAPPINGS = { 'id': dict_utils.id_name_dict, } SEARCH_FIELDS = () class Meta: """Model configuration.""" doc_type = 'relatedmodel'
class CompaniesHouseCompany(BaseESModel): """Elasticsearch representation of CompaniesHouseCompany model.""" id = Keyword() company_category = fields.SortableCaseInsensitiveKeywordText() company_number = fields.SortableCaseInsensitiveKeywordText() company_status = fields.SortableCaseInsensitiveKeywordText() incorporation_date = Date() name = fields.SortableText(copy_to=[ 'name_keyword', 'name_trigram', ], ) name_keyword = fields.SortableCaseInsensitiveKeywordText() name_trigram = fields.TrigramText() registered_address_1 = Text() registered_address_2 = Text() registered_address_town = fields.SortableCaseInsensitiveKeywordText() registered_address_county = Text() registered_address_postcode = Text( copy_to='registered_address_postcode_trigram') registered_address_postcode_trigram = fields.TrigramText() registered_address_country = fields.nested_id_name_field() sic_code_1 = Text() sic_code_2 = Text() sic_code_3 = Text() sic_code_4 = Text() uri = Text() MAPPINGS = { 'id': str, 'registered_address_country': dict_utils.id_name_dict, } SEARCH_FIELDS = ( # to match names like A & B 'name', 'name_trigram', 'company_number', 'registered_address_postcode_trigram', ) class Meta: """Default document meta data.""" doc_type = 'companieshousecompany'
class Order(BaseESModel): """Elasticsearch representation of Order model.""" id = Keyword() reference = fields.SortableCaseInsensitiveKeywordText( copy_to=['reference_trigram']) reference_trigram = fields.TrigramText() status = fields.SortableCaseInsensitiveKeywordText() company = fields.nested_company_field('company') contact = fields.nested_contact_or_adviser_field('contact') created_by = fields.nested_contact_or_adviser_field('created_by', include_dit_team=True) created_on = Date() modified_on = Date() primary_market = fields.nested_id_name_field() sector = fields.nested_sector_field() uk_region = fields.nested_id_name_field() description = fields.EnglishText() contacts_not_to_approach = Text() further_info = Text() existing_agents = Text(index=False) delivery_date = Date() service_types = fields.nested_id_name_field() contact_email = fields.SortableCaseInsensitiveKeywordText() contact_phone = Keyword() subscribers = fields.nested_contact_or_adviser_field('subscribers', include_dit_team=True) assignees = fields.nested_contact_or_adviser_field('assignees', include_dit_team=True) po_number = Keyword(index=False) discount_value = Integer(index=False) vat_status = Keyword(index=False) vat_number = Keyword(index=False) vat_verified = Boolean(index=False) net_cost = Integer(index=False) subtotal_cost_string = Keyword() subtotal_cost = Integer(copy_to=['subtotal_cost_string']) vat_cost = Integer(index=False) total_cost_string = Keyword() total_cost = Integer(copy_to=['total_cost_string']) payment_due_date = Date() paid_on = Date() completed_by = fields.nested_contact_or_adviser_field('completed_by') completed_on = Date() cancelled_by = fields.nested_contact_or_adviser_field('cancelled_by') cancelled_on = Date() cancellation_reason = fields.nested_id_name_field() billing_company_name = Text() billing_contact_name = Text() billing_email = fields.SortableCaseInsensitiveKeywordText() billing_phone = fields.SortableCaseInsensitiveKeywordText() billing_address_1 = Text() billing_address_2 = Text() billing_address_town = fields.SortableCaseInsensitiveKeywordText() billing_address_county = fields.SortableCaseInsensitiveKeywordText() billing_address_postcode = Text() billing_address_country = fields.nested_id_name_field() MAPPINGS = { 'id': str, 'company': dict_utils.company_dict, 'contact': dict_utils.contact_or_adviser_dict, 'created_by': dict_utils.adviser_dict_with_team, 'primary_market': dict_utils.id_name_dict, 'sector': dict_utils.sector_dict, 'uk_region': dict_utils.id_name_dict, 'service_types': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'subscribers': lambda col: [ dict_utils.contact_or_adviser_dict(c.adviser, include_dit_team=True) for c in col.all() ], 'assignees': lambda col: [ dict_utils.contact_or_adviser_dict(c.adviser, include_dit_team=True) for c in col.all() ], 'billing_address_country': dict_utils.id_name_dict, 'completed_by': dict_utils.contact_or_adviser_dict, 'cancelled_by': dict_utils.contact_or_adviser_dict, 'cancellation_reason': dict_utils.id_name_dict, } COMPUTED_MAPPINGS = { 'payment_due_date': lambda x: x.invoice.payment_due_date if x.invoice else None, } SEARCH_FIELDS = ( 'reference_trigram', 'company.name', 'company.name_trigram', 'contact.name', 'contact.name_trigram', 'total_cost_string', 'subtotal_cost_string', ) class Meta: """Default document meta data.""" doc_type = 'order'
class InvestmentProject(BaseESModel): """Elasticsearch representation of InvestmentProject.""" id = Keyword() actual_land_date = Date() actual_uk_regions = fields.nested_id_name_field() address_1 = Text() address_2 = Text() address_town = fields.SortableCaseInsensitiveKeywordText() address_postcode = Text() approved_commitment_to_invest = Boolean() approved_fdi = Boolean() approved_good_value = Boolean() approved_high_value = Boolean() approved_landed = Boolean() approved_non_fdi = Boolean() allow_blank_estimated_land_date = Boolean(index=False) allow_blank_possible_uk_regions = Boolean(index=False) anonymous_description = fields.EnglishText() archived = Boolean() archived_by = fields.nested_contact_or_adviser_field('archived_by') archived_on = Date() archived_reason = Text() associated_non_fdi_r_and_d_project = _nested_investment_project_field() average_salary = fields.nested_id_name_field() business_activities = fields.nested_id_name_field() client_cannot_provide_foreign_investment = Boolean() client_cannot_provide_total_investment = Boolean() client_contacts = fields.nested_contact_or_adviser_field('client_contacts') client_relationship_manager = fields.nested_contact_or_adviser_field( 'client_relationship_manager', include_dit_team=True, ) client_requirements = fields.TextWithKeyword() comments = fields.EnglishText() country_lost_to = _country_lost_to_mapping() created_on = Date() created_by = fields.nested_contact_or_adviser_field( 'created_by', include_dit_team=True, ) date_abandoned = Date() date_lost = Date() delivery_partners = fields.nested_id_name_field() description = fields.EnglishText() estimated_land_date = Date() export_revenue = Boolean() fdi_type = fields.nested_id_name_field() fdi_value = fields.nested_id_name_field() foreign_equity_investment = Double() government_assistance = Boolean() intermediate_company = fields.nested_id_name_field() investor_company = fields.nested_id_name_partial_field('investor_company') investor_company_country = fields.nested_id_name_field() investment_type = fields.nested_id_name_field() investor_type = fields.nested_id_name_field() level_of_involvement = fields.nested_id_name_field() likelihood_of_landing = Long() project_assurance_adviser = fields.nested_contact_or_adviser_field( 'project_assurance_adviser', include_dit_team=True, ) project_manager = fields.nested_contact_or_adviser_field( 'project_manager', include_dit_team=True, ) name = fields.SortableText(copy_to=['name_keyword', 'name_trigram']) name_keyword = fields.SortableCaseInsensitiveKeywordText() name_trigram = fields.TrigramText() new_tech_to_uk = Boolean() non_fdi_r_and_d_budget = Boolean() number_new_jobs = Integer() number_safeguarded_jobs = Long() modified_on = Date() project_arrived_in_triage_on = Date() project_code = fields.SortableCaseInsensitiveKeywordText( copy_to='project_code_trigram') project_code_trigram = fields.TrigramText() proposal_deadline = Date() other_business_activity = fields.TextWithKeyword() quotable_as_public_case_study = Boolean() r_and_d_budget = Boolean() reason_abandoned = fields.TextWithKeyword() reason_delayed = fields.TextWithKeyword() reason_lost = fields.TextWithKeyword() referral_source_activity = fields.nested_id_name_field() referral_source_activity_event = fields.SortableCaseInsensitiveKeywordText( ) referral_source_activity_marketing = fields.nested_id_name_field() referral_source_activity_website = fields.nested_id_name_field() referral_source_adviser = _referral_source_adviser_mapping() sector = fields.nested_sector_field() site_decided = Boolean() some_new_jobs = Boolean() specific_programme = fields.nested_id_name_field() stage = fields.nested_id_name_field() status = fields.SortableCaseInsensitiveKeywordText() team_members = fields.nested_contact_or_adviser_field( 'team_members', include_dit_team=True) total_investment = Double() uk_company = fields.nested_id_name_partial_field('uk_company') uk_company_decided = Boolean() uk_region_locations = fields.nested_id_name_field() will_new_jobs_last_two_years = Boolean() MAPPINGS = { 'id': str, 'actual_uk_regions': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'archived_by': dict_utils.contact_or_adviser_dict, 'associated_non_fdi_r_and_d_project': dict_utils.investment_project_dict, 'average_salary': dict_utils.id_name_dict, 'business_activities': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'client_contacts': lambda col: [dict_utils.contact_or_adviser_dict(c) for c in col.all()], 'client_relationship_manager': dict_utils.adviser_dict_with_team, 'country_lost_to': dict_utils.id_name_dict, 'created_by': dict_utils.adviser_dict_with_team, 'delivery_partners': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'fdi_type': dict_utils.id_name_dict, 'fdi_value': dict_utils.id_name_dict, 'intermediate_company': dict_utils.id_name_dict, 'investment_type': dict_utils.id_name_dict, 'investor_company': dict_utils.id_name_dict, 'investor_type': dict_utils.id_name_dict, 'level_of_involvement': dict_utils.id_name_dict, 'project_assurance_adviser': dict_utils.adviser_dict_with_team, 'project_code': str, 'project_manager': dict_utils.adviser_dict_with_team, 'referral_source_activity': dict_utils.id_name_dict, 'referral_source_activity_marketing': dict_utils.id_name_dict, 'referral_source_activity_website': dict_utils.id_name_dict, 'referral_source_adviser': dict_utils.contact_or_adviser_dict, 'sector': dict_utils.sector_dict, 'specific_programme': dict_utils.id_name_dict, 'stage': dict_utils.id_name_dict, 'team_members': lambda col: [ dict_utils.contact_or_adviser_dict(c.adviser, include_dit_team=True) for c in col.all() ], 'uk_company': dict_utils.id_name_dict, 'uk_region_locations': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], } COMPUTED_MAPPINGS = { 'investor_company_country': dict_utils.computed_nested_id_name_dict( 'investor_company.registered_address_country', ), } SEARCH_FIELDS = ( 'name', 'name_trigram', 'uk_company.name', 'uk_company.name_trigram', 'investor_company.name', 'investor_company.name_trigram', 'project_code_trigram', ) class Meta: """Default document meta data.""" doc_type = 'investment_project'
class Company(BaseESModel): """Elasticsearch representation of Company model.""" id = Keyword() archived = Boolean() archived_by = fields.nested_contact_or_adviser_field('archived_by') archived_on = Date() archived_reason = Text() business_type = fields.nested_id_name_field() classification = fields.nested_id_name_field() companies_house_data = fields.nested_ch_company_field() company_number = fields.SortableCaseInsensitiveKeywordText() contacts = fields.nested_contact_or_adviser_field('contacts') created_on = Date() description = fields.EnglishText() employee_range = fields.nested_id_name_field() export_experience_category = fields.nested_id_name_field() export_to_countries = fields.nested_id_name_field() future_interest_countries = fields.nested_id_name_field() global_headquarters = fields.nested_id_name_field() headquarter_type = fields.nested_id_name_field() modified_on = Date() name = fields.SortableText(copy_to=['name_keyword', 'name_trigram']) name_keyword = fields.SortableCaseInsensitiveKeywordText() name_trigram = fields.TrigramText() one_list_account_owner = fields.nested_contact_or_adviser_field('one_list_account_owner') reference_code = fields.SortableCaseInsensitiveKeywordText() registered_address_1 = Text() registered_address_2 = Text() registered_address_town = fields.SortableCaseInsensitiveKeywordText() registered_address_county = Text() registered_address_country = fields.nested_id_name_partial_field( 'registered_address_country', ) registered_address_postcode = Text( copy_to=[ 'registered_address_postcode_trigram', ], ) registered_address_postcode_trigram = fields.TrigramText() sector = fields.nested_sector_field() trading_address_1 = Text() trading_address_2 = Text() trading_address_town = fields.SortableCaseInsensitiveKeywordText() trading_address_county = Text() trading_address_postcode = Text( copy_to=['trading_address_postcode_trigram'], ) trading_address_postcode_trigram = fields.TrigramText() trading_address_country = fields.nested_id_name_partial_field( 'trading_address_country', ) trading_name = fields.SortableText( copy_to=[ 'trading_name_keyword', 'trading_name_trigram', ], ) trading_name_keyword = fields.SortableCaseInsensitiveKeywordText() trading_name_trigram = fields.TrigramText() turnover_range = fields.nested_id_name_field() uk_region = fields.nested_id_name_field() uk_based = Boolean() vat_number = Keyword(index=False) website = Text() COMPUTED_MAPPINGS = { 'trading_name': attrgetter('alias'), } MAPPINGS = { 'id': str, 'archived_by': dict_utils.contact_or_adviser_dict, 'business_type': dict_utils.id_name_dict, 'classification': dict_utils.id_name_dict, 'companies_house_data': dict_utils.ch_company_dict, 'contacts': lambda col: [dict_utils.contact_or_adviser_dict(c) for c in col.all()], 'employee_range': dict_utils.id_name_dict, 'export_experience_category': dict_utils.id_name_dict, 'export_to_countries': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'future_interest_countries': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'global_headquarters': dict_utils.id_name_dict, 'headquarter_type': dict_utils.id_name_dict, 'one_list_account_owner': dict_utils.contact_or_adviser_dict, 'registered_address_country': dict_utils.id_name_dict, 'sector': dict_utils.sector_dict, 'trading_address_country': dict_utils.id_name_dict, 'turnover_range': dict_utils.id_name_dict, 'uk_based': bool, 'uk_region': dict_utils.id_name_dict, } SEARCH_FIELDS = ( 'name', 'name_trigram', 'company_number', 'trading_name', 'trading_name_trigram', 'reference_code', 'registered_address_country.name_trigram', 'registered_address_postcode_trigram', 'trading_address_country.name_trigram', 'trading_address_postcode_trigram', 'uk_region.name_trigram', ) class Meta: """Default document meta data.""" doc_type = 'company'
class Contact(BaseESModel): """Elasticsearch representation of Contact model.""" id = Keyword() accepts_dit_email_marketing = Boolean() address_1 = Text() address_2 = Text() address_town = fields.SortableCaseInsensitiveKeywordText() address_county = fields.SortableCaseInsensitiveKeywordText() address_postcode = Text() address_country = fields.nested_id_name_field() address_same_as_company = Boolean() adviser = fields.nested_contact_or_adviser_field('adviser') archived = Boolean() archived_by = fields.nested_contact_or_adviser_field('archived_by') archived_on = Date() archived_reason = Text() company = fields.nested_company_field('company') company_sector = fields.nested_sector_field() company_uk_region = fields.nested_id_name_field() created_by = fields.nested_contact_or_adviser_field('created_by', include_dit_team=True) created_on = Date() email = fields.SortableCaseInsensitiveKeywordText() email_alternative = Text() first_name = fields.SortableText( copy_to=[ 'name', 'name_keyword', 'name_trigram', ], ) job_title = fields.SortableCaseInsensitiveKeywordText() last_name = fields.SortableText( copy_to=[ 'name', 'name_keyword', 'name_trigram', ], ) modified_on = Date() name = fields.SortableText() name_keyword = fields.SortableCaseInsensitiveKeywordText() # field is being aggregated name_trigram = fields.TrigramText() notes = fields.EnglishText() primary = Boolean() telephone_alternative = Text() telephone_countrycode = Keyword() telephone_number = Keyword() title = fields.nested_id_name_field() MAPPINGS = { 'id': str, 'adviser': dict_utils.contact_or_adviser_dict, 'archived_by': dict_utils.contact_or_adviser_dict, 'company': dict_utils.company_dict, 'created_by': dict_utils.adviser_dict_with_team, 'title': dict_utils.id_name_dict, } COMPUTED_MAPPINGS = { 'address_1': contact_dict_utils.computed_address_field('address_1'), 'address_2': contact_dict_utils.computed_address_field('address_2'), 'address_town': contact_dict_utils.computed_address_field('address_town'), 'address_county': contact_dict_utils.computed_address_field('address_county'), 'address_postcode': contact_dict_utils.computed_address_field('address_postcode'), 'address_country': contact_dict_utils.computed_address_field('address_country'), 'company_sector': dict_utils.computed_nested_sector_dict('company.sector'), 'company_uk_region': dict_utils.computed_nested_id_name_dict('company.uk_region'), } SEARCH_FIELDS = ( 'name', 'name_trigram', 'email', 'email_alternative', 'company.name', 'company.name_trigram', ) class Meta: """Default document meta data.""" doc_type = 'contact'
class Event(BaseESModel): """Elasticsearch representation of Event model.""" id = Keyword() address_1 = Text() address_2 = Text() address_town = fields.SortableCaseInsensitiveKeywordText() address_county = fields.SortableCaseInsensitiveKeywordText() address_postcode = Text(copy_to='address_postcode_trigram') address_postcode_trigram = fields.TrigramText() address_country = fields.nested_id_name_partial_field('address_country') created_on = Date() disabled_on = Date() end_date = Date() event_type = fields.nested_id_name_field() lead_team = fields.nested_id_name_field() location_type = fields.nested_id_name_field() modified_on = Date() name = fields.SortableText(copy_to=['name_keyword', 'name_trigram']) name_keyword = fields.SortableCaseInsensitiveKeywordText() name_trigram = fields.TrigramText() notes = fields.EnglishText() organiser = fields.nested_contact_or_adviser_field('organiser') related_programmes = fields.nested_id_name_partial_field('related_programmes') service = fields.nested_id_name_field() start_date = Date() teams = fields.nested_id_name_partial_field('teams') uk_region = fields.nested_id_name_partial_field('uk_region') MAPPINGS = { 'id': str, 'address_country': dict_utils.id_name_dict, 'event_type': dict_utils.id_name_dict, 'lead_team': dict_utils.id_name_dict, 'location_type': dict_utils.id_name_dict, 'organiser': dict_utils.contact_or_adviser_dict, 'related_programmes': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'service': dict_utils.id_name_dict, 'teams': lambda col: [dict_utils.id_name_dict(c) for c in col.all()], 'uk_region': dict_utils.id_name_dict, } COMPUTED_MAPPINGS = {} SEARCH_FIELDS = ( 'name', 'name_trigram', 'address_country.name_trigram', 'address_postcode_trigram', 'uk_region.name_trigram', 'organiser.name_trigram', 'teams.name', 'teams.name_trigram', 'related_programmes.name', 'related_programmes.name_trigram', ) class Meta: """Default document meta data.""" doc_type = 'event'
class Interaction(BaseESModel): """Elasticsearch representation of Interaction model.""" id = Keyword() company = fields.nested_company_field('company') company_sector = fields.nested_sector_field() communication_channel = fields.nested_id_name_field() contact = fields.nested_contact_or_adviser_field('contact') created_on = Date() date = Date() dit_adviser = fields.nested_contact_or_adviser_field('dit_adviser') dit_team = fields.nested_id_name_partial_field('dit_team') event = fields.nested_id_name_partial_field('event') investment_project = fields.nested_id_name_field() investment_project_sector = fields.nested_sector_field() is_event = Boolean() grant_amount_offered = Double() kind = Keyword() modified_on = Date() net_company_receipt = Double() notes = fields.EnglishText() service = fields.nested_id_name_field() service_delivery_status = fields.nested_id_name_field() subject = fields.SortableCaseInsensitiveKeywordText( copy_to=['subject_english'], ) subject_english = fields.EnglishText() MAPPINGS = { 'id': str, 'company': dict_utils.company_dict, 'communication_channel': dict_utils.id_name_dict, 'contact': dict_utils.contact_or_adviser_dict, 'dit_adviser': dict_utils.contact_or_adviser_dict, 'dit_team': dict_utils.id_name_dict, 'event': dict_utils.id_name_dict, 'investment_project': dict_utils.id_name_dict, 'service': dict_utils.id_name_dict, 'service_delivery_status': dict_utils.id_name_dict, } COMPUTED_MAPPINGS = { 'company_sector': dict_utils.computed_nested_sector_dict('company.sector'), 'investment_project_sector': dict_utils.computed_nested_sector_dict('investment_project.sector', ), 'is_event': attrgetter('is_event'), } SEARCH_FIELDS = ( 'company.name', 'company.name_trigram', 'contact.name', 'contact.name_trigram', 'event.name', 'event.name_trigram', 'subject_english', 'dit_adviser.name', 'dit_adviser.name_trigram', 'dit_team.name', 'dit_team.name_trigram', ) class Meta: """Default document meta data.""" doc_type = 'interaction'