85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
|
from typing import Dict, List, Any
|
||
|
|
from .base import get_db
|
||
|
|
|
||
|
|
class ARFieldModel:
|
||
|
|
"""Model for custom AR field operations"""
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_all(tenant_id: str) -> List[Dict[str, Any]]:
|
||
|
|
"""Get all custom AR fields for a tenant"""
|
||
|
|
tenant_id = tenant_id.lower()
|
||
|
|
|
||
|
|
with get_db() as conn:
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute('''
|
||
|
|
SELECT id, field_name, label, field_type, editable, display_order
|
||
|
|
FROM custom_ar_fields
|
||
|
|
WHERE tenant_id = ?
|
||
|
|
ORDER BY display_order, id
|
||
|
|
''', (tenant_id,))
|
||
|
|
|
||
|
|
fields = []
|
||
|
|
for row in cursor.fetchall():
|
||
|
|
fields.append({
|
||
|
|
'id': row['id'],
|
||
|
|
'fieldName': row['field_name'],
|
||
|
|
'label': row['label'],
|
||
|
|
'fieldType': row['field_type'],
|
||
|
|
'editable': row['editable'],
|
||
|
|
'displayOrder': row['display_order']
|
||
|
|
})
|
||
|
|
|
||
|
|
return fields
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def save(tenant_id: str, field_data: Dict[str, Any]):
|
||
|
|
"""Save or update a custom AR field"""
|
||
|
|
tenant_id = tenant_id.lower()
|
||
|
|
|
||
|
|
with get_db() as conn:
|
||
|
|
cursor = conn.cursor()
|
||
|
|
|
||
|
|
if 'id' in field_data and field_data['id']:
|
||
|
|
# Update existing field
|
||
|
|
cursor.execute('''
|
||
|
|
UPDATE custom_ar_fields
|
||
|
|
SET field_name = ?, label = ?, field_type = ?, editable = ?,
|
||
|
|
display_order = ?, updated_at = CURRENT_TIMESTAMP
|
||
|
|
WHERE id = ? AND tenant_id = ?
|
||
|
|
''', (
|
||
|
|
field_data['fieldName'],
|
||
|
|
field_data['label'],
|
||
|
|
field_data['fieldType'],
|
||
|
|
field_data['editable'],
|
||
|
|
field_data['displayOrder'],
|
||
|
|
field_data['id'],
|
||
|
|
tenant_id
|
||
|
|
))
|
||
|
|
else:
|
||
|
|
# Insert new field
|
||
|
|
cursor.execute('''
|
||
|
|
INSERT INTO custom_ar_fields
|
||
|
|
(tenant_id, field_name, label, field_type, editable, display_order)
|
||
|
|
VALUES (?, ?, ?, ?, ?, ?)
|
||
|
|
''', (
|
||
|
|
tenant_id,
|
||
|
|
field_data['fieldName'],
|
||
|
|
field_data['label'],
|
||
|
|
field_data['fieldType'],
|
||
|
|
field_data['editable'],
|
||
|
|
field_data['displayOrder']
|
||
|
|
))
|
||
|
|
|
||
|
|
conn.commit()
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def delete(tenant_id: str, field_id: int):
|
||
|
|
"""Delete a custom AR field"""
|
||
|
|
tenant_id = tenant_id.lower()
|
||
|
|
|
||
|
|
with get_db() as conn:
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute('DELETE FROM custom_ar_fields WHERE id = ? AND tenant_id = ?',
|
||
|
|
(field_id, tenant_id))
|
||
|
|
conn.commit()
|