Files
westech-r2/westech_r2/add_template_links.py
T

86 lines
3.2 KiB
Python

import frappe
frappe.init(site="erpnext.local")
frappe.connect()
try:
# Get Westech workspace
ws = frappe.get_doc("Workspace", "Westech")
# Create HTML block with template links
html_block = '''
<div style="border: 2px solid #6f42c1; border-radius: 8px; padding: 15px; margin: 10px 0;">
<h4 style="margin-top: 0; color: #6f42c1;">📋 Print Format Templates</h4>
<div style="display: grid; grid-template-columns: repeat(2, 1fr); gap: 10px;">
<a href="/app/print-format/special-handling-log" target="_blank"
style="display: block; padding: 10px; background: #f0f0f0; border-radius: 5px; text-decoration: none; color: #333;">
<strong>🟣 Purple Sheet</strong><br>
<small>Special Handling Log</small>
</a>
<a href="/app/print-format/r2-special-handling-log" target="_blank"
style="display: block; padding: 10px; background: #e8f5e9; border-radius: 5px; text-decoration: none; color: #333;">
<strong>🟢 Green Sheet</strong><br>
<small>R2 Special Handling Log</small>
</a>
<a href="/app/print-format/pallet-label-4x6" target="_blank"
style="display: block; padding: 10px; background: #fff3cd; border-radius: 5px; text-decoration: none; color: #333;">
<strong>🏷️ Pallet Label</strong><br>
<small>4x6" with Service Level</small>
</a>
<a href="/app/print-format/r2-uw-label" target="_blank"
style="display: block; padding: 10px; background: #e3f2fd; border-radius: 5px; text-decoration: none; color: #333;">
<strong>📦 R2 UW Label</strong><br>
<small>Controlled Stream</small>
</a>
</div>
</div>
'''
# Parse existing content
import json
content = json.loads(ws.content) if ws.content else []
# Check if template links section already exists
existing_idx = None
for i, block in enumerate(content):
if block.get('type') == 'custom_block' and 'Print Format Templates' in block.get('data', {}).get('html', ''):
existing_idx = i
break
if existing_idx is not None:
# Update existing block
content[existing_idx]['data']['html'] = html_block
print(f"✓ Updated existing template links block")
else:
# Add new block after header
template_block = {
"type": "custom_block",
"data": {
"html": html_block
}
}
# Insert after first header block
insert_idx = 1
for i, block in enumerate(content):
if block.get('type') == 'header':
insert_idx = i + 1
break
content.insert(insert_idx, template_block)
print(f"✓ Added new template links block")
# Update workspace
ws.content = json.dumps(content)
# Remove broken shortcuts before saving
if hasattr(ws, 'shortcuts'):
ws.shortcuts = [s for s in ws.shortcuts if frappe.db.exists(s.doctype, s.link_to)]
ws.save(ignore_permissions=True)
frappe.db.commit()
print(f"✓ Workspace 'Westech' updated successfully")
print(f"✓ View at: https://erp.diagalon.com/app/westech")
finally:
frappe.destroy()