inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-11-04 07:05:41 +00:00
Code Activity

So I learned something today...

Browse Source 
In migration files you can access the "historical" pythonic model, and use that, with *all* the helpers,
rather than writing clunky old SQL!!!!

:'(
This commit is contained in:
Oliver Walters
2021-02-04 23:10:10 +11:00
parent 74704a7c1e
commit 3c5169c793
1 changed files with 9 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
InvenTree/company/migrations/0019_auto_20200413_0642.py
View File

@@ -165,24 +165,19 @@ def associate_manufacturers(apps, schema_editor):
def create_manufacturer(part_id, input_name, company_name):
""" Create a new manufacturer """
# Manually create a new database row
# Note: Have to fill out all empty string values!
new_manufacturer_query = f"insert into company_company (name, description, is_customer, is_supplier, is_manufacturer, address, website, phone, email, contact, link, notes) values ('{company_name}', '{company_name}', 0, 0, 1, '', '', '', '', '', '', '');"
Company = apps.get_model('company', 'company')
cursor = connection.cursor()
cursor.execute(new_manufacturer_query)
# Extract the company back from the database
response = cursor.execute(f"select id from company_company where name='{company_name}';")
row = cursor.fetchone()
manufacturer_id = int(row[0])
manufacturer = Company.objects.create(
name=company_name,
description=company_name,
is_manufacturer=True
)
# Map both names to the same company
links[input_name] = manufacturer_id
links[company_name] = manufacturer_id
links[input_name] = manufacturer.pk
links[company_name] = manufacturer.pk
companies[company_name] = manufacturer_id
companies[company_name] = manufacturer.pk
print(" - Part[{pk}]: Created new manufacturer: '{name}'".format(pk=part_id, name=company_name))