def company_model(self): return get_company_model()
def save(self, *args, **kwargs): if self.company: self.company.save() else: self.company = get_company_model().objects.create(denomination=self.denomination, since=self.since) super().save()
from django import forms from company import get_company_model, backends_loop, create_company CompanyModel = get_company_model() class CompanySearchByCountryForm(forms.Form): search = forms.CharField(required=True) class CompanyAddByCountry(forms.ModelForm): position = forms.IntegerField(required=True) search = forms.CharField(required=True) error_messages = { 'invalid_search': 'test', } def __init__(self, country_model, country_fields, parent_object=None, admin=False, *args, **kwargs): self.country_model = country_model self.country_fields = country_fields self.parent_object = parent_object super().__init__(*args, **kwargs) def get_results(self, country, search): message, companies, total, pages = backends_loop(country, search) return {
def get_country_model(self): return get_company_model( getattr(conf.Model, 'Company%s' % self.get_country().upper()))
class CompanyCheckSiren(CheckData): model = get_company_model() test_field = 'company_fr__siren'
class CompanyCheckRna(CheckData): model = get_company_model() test_field = 'company_fr__rna'