Esempio n. 1
0
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'
Esempio n. 2
0
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'
Esempio n. 3
0
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'
Esempio n. 4
0
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'
Esempio n. 5
0
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'
Esempio n. 6
0
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'
Esempio n. 7
0
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'
Esempio n. 8
0
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'