chore(seed): add pro appliance demo data generator
This commit is contained in:
Ruslan Bakiev
319
odoo/scripts/generate_pro_appliance_demo.py
Normal file
319
odoo/scripts/generate_pro_appliance_demo.py
Normal file
@@ -0,0 +1,319 @@
|
|||||||
|
# Run with:
|
||||||
|
# odoo shell -c /etc/odoo/odoo.conf -d <db_name> < odoo/scripts/generate_pro_appliance_demo.py
|
||||||
|
|
||||||
|
from datetime import timedelta
|
||||||
|
|
||||||
|
from odoo import fields
|
||||||
|
|
||||||
|
|
||||||
|
SEED_TAG = "[SEED:pro-appliance.repair]"
|
||||||
|
SOURCE_URL = "https://pro-appliance.repair/"
|
||||||
|
|
||||||
|
|
||||||
|
CONTACTS = [
|
||||||
|
{"name": "Christina Gomez", "phone": "+1-302-503-0047"},
|
||||||
|
{"name": "Clinton Parker", "phone": "+1-302-503-0048"},
|
||||||
|
{"name": "Mike Kreller", "phone": "+1-302-503-0049"},
|
||||||
|
{"name": "Frank Smith", "phone": "+1-302-503-0050"},
|
||||||
|
{"name": "Mandy Cekine", "phone": "+1-302-503-0051"},
|
||||||
|
]
|
||||||
|
|
||||||
|
LOCATIONS = [
|
||||||
|
{
|
||||||
|
"description": "Newark Downtown, DE 19713",
|
||||||
|
"latitude": 39.683723,
|
||||||
|
"longitude": -75.749657,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Christiana Area, Newark, DE",
|
||||||
|
"latitude": 39.678002,
|
||||||
|
"longitude": -75.659408,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Bear Area, New Castle County, DE",
|
||||||
|
"latitude": 39.629276,
|
||||||
|
"longitude": -75.658262,
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
WORK_TYPES = [
|
||||||
|
("Clothes Dryer", 90),
|
||||||
|
("Dishwasher", 120),
|
||||||
|
("Range", 120),
|
||||||
|
("Refrigerator", 120),
|
||||||
|
("Microwave", 90),
|
||||||
|
]
|
||||||
|
|
||||||
|
FALLBACK_MATERIALS = [
|
||||||
|
("Dryer Belt", "DRY-BELT", 35.0),
|
||||||
|
("Dishwasher Drain Pump", "DW-PUMP", 72.0),
|
||||||
|
("Range Igniter", "RANGE-IGN", 48.0),
|
||||||
|
("Refrigerator Thermistor", "FRIDGE-THRM", 29.0),
|
||||||
|
("Microwave Door Switch", "MW-SWITCH", 17.0),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_source():
|
||||||
|
source_model = env["contact.source"].sudo()
|
||||||
|
source = source_model.search([("name", "=", "pro-appliance.repair Website")], limit=1)
|
||||||
|
if not source:
|
||||||
|
source = source_model.create(
|
||||||
|
{
|
||||||
|
"name": "pro-appliance.repair Website",
|
||||||
|
"description": f"Generated from {SOURCE_URL} {SEED_TAG}",
|
||||||
|
"state": "active",
|
||||||
|
"active": True,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
return source
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_phone_type():
|
||||||
|
phone_type = env["dsrpt.communication.type"].sudo().search([("code", "=", "phone")], limit=1)
|
||||||
|
return phone_type
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_work_types():
|
||||||
|
model = env["repair.work.type"].sudo()
|
||||||
|
result = []
|
||||||
|
for name, duration in WORK_TYPES:
|
||||||
|
rec = model.search([("name", "=", name)], limit=1)
|
||||||
|
if not rec:
|
||||||
|
rec = model.create({"name": name, "duration_min": duration, "active": True})
|
||||||
|
else:
|
||||||
|
values = {}
|
||||||
|
if not rec.duration_min:
|
||||||
|
values["duration_min"] = duration
|
||||||
|
if not rec.active:
|
||||||
|
values["active"] = True
|
||||||
|
if values:
|
||||||
|
rec.write(values)
|
||||||
|
result.append(rec)
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_materials():
|
||||||
|
material_model = env["repair.material"].sudo()
|
||||||
|
materials = material_model.search([("state", "=", "active")], order="id", limit=20)
|
||||||
|
if materials:
|
||||||
|
return materials
|
||||||
|
|
||||||
|
existing_any = material_model.search([], order="id", limit=20)
|
||||||
|
if existing_any:
|
||||||
|
for rec in existing_any.filtered(lambda m: m.state != "active"):
|
||||||
|
rec.action_set_active()
|
||||||
|
return material_model.search([("state", "=", "active")], order="id", limit=20)
|
||||||
|
|
||||||
|
created = material_model.browse()
|
||||||
|
for name, code, cost in FALLBACK_MATERIALS:
|
||||||
|
rec = material_model.create(
|
||||||
|
{
|
||||||
|
"name": name,
|
||||||
|
"default_code": code,
|
||||||
|
"uom_name": "pcs",
|
||||||
|
"standard_cost": cost,
|
||||||
|
"state": "active",
|
||||||
|
"active": True,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
created |= rec
|
||||||
|
return created
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_technician(work_types):
|
||||||
|
technician_model = env["repair.technician"].sudo()
|
||||||
|
zone_model = env["repair.fsm.zone"].sudo()
|
||||||
|
|
||||||
|
technician = technician_model.search([("state", "=", "active")], limit=1, order="id")
|
||||||
|
if technician:
|
||||||
|
missing_types = work_types - technician.work_type_ids
|
||||||
|
if missing_types:
|
||||||
|
technician.write({"work_type_ids": [(4, wt.id) for wt in missing_types]})
|
||||||
|
return technician
|
||||||
|
|
||||||
|
zones = zone_model.search([("state", "=", "active")], limit=3)
|
||||||
|
technician = technician_model.create(
|
||||||
|
{
|
||||||
|
"name": "Alex Pro Appliance",
|
||||||
|
"state": "active",
|
||||||
|
"active": True,
|
||||||
|
"work_type_ids": [(6, 0, work_types.ids)],
|
||||||
|
"zone_ids": [(6, 0, zones.ids)],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
schedule_model = env["repair.technician.schedule"].sudo()
|
||||||
|
for day in ["0", "1", "2", "3", "4", "5"]:
|
||||||
|
schedule_model.create(
|
||||||
|
{
|
||||||
|
"technician_id": technician.id,
|
||||||
|
"day_of_week": day,
|
||||||
|
"hour_from": 9.0,
|
||||||
|
"hour_to": 17.0,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
return technician
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_contact_and_address(payload, location, source, phone_type):
|
||||||
|
contact_model = env["dsrpt.contact"].sudo()
|
||||||
|
address_model = env["dsrpt.contact.address"].sudo()
|
||||||
|
comm_model = env["dsrpt.contact.communication"].sudo()
|
||||||
|
|
||||||
|
contact = contact_model.search([("name", "=", payload["name"])], limit=1)
|
||||||
|
if not contact:
|
||||||
|
contact = contact_model.create(
|
||||||
|
{
|
||||||
|
"name": payload["name"],
|
||||||
|
"status": "has_request",
|
||||||
|
"source_id": source.id,
|
||||||
|
"note": f"Imported from {SOURCE_URL} {SEED_TAG}",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
address = address_model.search(
|
||||||
|
[
|
||||||
|
("contact_id", "=", contact.id),
|
||||||
|
("description", "=", location["description"]),
|
||||||
|
],
|
||||||
|
limit=1,
|
||||||
|
)
|
||||||
|
if not address:
|
||||||
|
address = address_model.create(
|
||||||
|
{
|
||||||
|
"contact_id": contact.id,
|
||||||
|
"description": location["description"],
|
||||||
|
"latitude": location["latitude"],
|
||||||
|
"longitude": location["longitude"],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
if phone_type:
|
||||||
|
existing_phone = comm_model.search(
|
||||||
|
[
|
||||||
|
("contact_id", "=", contact.id),
|
||||||
|
("communication_type_id", "=", phone_type.id),
|
||||||
|
("value", "=", payload["phone"]),
|
||||||
|
],
|
||||||
|
limit=1,
|
||||||
|
)
|
||||||
|
if not existing_phone:
|
||||||
|
comm_model.create(
|
||||||
|
{
|
||||||
|
"contact_id": contact.id,
|
||||||
|
"communication_type_id": phone_type.id,
|
||||||
|
"value": payload["phone"],
|
||||||
|
"is_preferred": True if not contact.communication_ids else False,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
return contact, address
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_work_order(contact, address, work_type, technician, materials, index):
|
||||||
|
order_model = env["repair.work.order"].sudo()
|
||||||
|
time_model = env["repair.work.order.time"].sudo()
|
||||||
|
mat_line_model = env["repair.work.order.material"].sudo()
|
||||||
|
|
||||||
|
desc = f"{SEED_TAG} {work_type.name} service request from {SOURCE_URL}"
|
||||||
|
existing = order_model.search(
|
||||||
|
[
|
||||||
|
("contact_id", "=", contact.id),
|
||||||
|
("work_type_id", "=", work_type.id),
|
||||||
|
("description", "=", desc),
|
||||||
|
],
|
||||||
|
limit=1,
|
||||||
|
)
|
||||||
|
if existing:
|
||||||
|
return existing, False
|
||||||
|
|
||||||
|
start_dt = fields.Datetime.now() + timedelta(days=index + 1, hours=index)
|
||||||
|
duration_min = max(work_type.duration_min or 90, 30)
|
||||||
|
end_dt = start_dt + timedelta(minutes=duration_min)
|
||||||
|
|
||||||
|
order = order_model.create(
|
||||||
|
{
|
||||||
|
"contact_id": contact.id,
|
||||||
|
"contact_address_id": address.id,
|
||||||
|
"work_type_id": work_type.id,
|
||||||
|
"description": desc,
|
||||||
|
"requested_datetime": fields.Datetime.now(),
|
||||||
|
"scheduled_datetime": start_dt,
|
||||||
|
"scheduled_end": end_dt,
|
||||||
|
"slot_day": fields.Date.to_date(start_dt),
|
||||||
|
"technician_id": technician.id if technician else False,
|
||||||
|
"state": "assigned" if technician else "confirmed",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
time_model.create(
|
||||||
|
{
|
||||||
|
"work_order_id": order.id,
|
||||||
|
"technician_id": technician.id if technician else False,
|
||||||
|
"description": f"{work_type.name} diagnostics and repair",
|
||||||
|
"hours": round(duration_min / 60.0, 2),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
if materials:
|
||||||
|
mat1 = materials[index % len(materials)]
|
||||||
|
mat_line_model.create(
|
||||||
|
{
|
||||||
|
"work_order_id": order.id,
|
||||||
|
"material_id": mat1.id,
|
||||||
|
"qty": 1.0 + (index % 2),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
if len(materials) > 1:
|
||||||
|
mat2 = materials[(index + 1) % len(materials)]
|
||||||
|
mat_line_model.create(
|
||||||
|
{
|
||||||
|
"work_order_id": order.id,
|
||||||
|
"material_id": mat2.id,
|
||||||
|
"qty": 1.0,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
return order, True
|
||||||
|
|
||||||
|
|
||||||
|
def run():
|
||||||
|
source = ensure_source()
|
||||||
|
phone_type = ensure_phone_type()
|
||||||
|
work_types = ensure_work_types()
|
||||||
|
materials = ensure_materials()
|
||||||
|
technician = ensure_technician(env["repair.work.type"].sudo().browse([wt.id for wt in work_types]))
|
||||||
|
|
||||||
|
created_contacts = 0
|
||||||
|
created_addresses = 0
|
||||||
|
created_orders = 0
|
||||||
|
|
||||||
|
for idx, payload in enumerate(CONTACTS):
|
||||||
|
location = LOCATIONS[idx % len(LOCATIONS)]
|
||||||
|
contact_before = env["dsrpt.contact"].sudo().search_count([("name", "=", payload["name"])])
|
||||||
|
address_before = env["dsrpt.contact.address"].sudo().search_count(
|
||||||
|
[("description", "=", location["description"])]
|
||||||
|
)
|
||||||
|
|
||||||
|
contact, address = ensure_contact_and_address(payload, location, source, phone_type)
|
||||||
|
if contact_before == 0:
|
||||||
|
created_contacts += 1
|
||||||
|
if address_before == 0:
|
||||||
|
created_addresses += 1
|
||||||
|
|
||||||
|
work_type = work_types[idx % len(work_types)]
|
||||||
|
_order, was_created = ensure_work_order(contact, address, work_type, technician, materials, idx)
|
||||||
|
if was_created:
|
||||||
|
created_orders += 1
|
||||||
|
|
||||||
|
env.cr.commit()
|
||||||
|
|
||||||
|
print("=== Pro Appliance demo seed completed ===")
|
||||||
|
print(f"Source: {SOURCE_URL}")
|
||||||
|
print(f"Contacts created: {created_contacts}")
|
||||||
|
print(f"Addresses created: {created_addresses}")
|
||||||
|
print(f"Work orders created: {created_orders}")
|
||||||
|
print(f"Materials available for linking: {len(materials)}")
|
||||||
|
|
||||||
|
|
||||||
|
run()
|
||||||
Reference in New Issue
Block a user