alpha v0.938
This commit is contained in:
286
handlers/computers.py
Normal file
286
handlers/computers.py
Normal file
@@ -0,0 +1,286 @@
|
||||
from flask import flash, redirect, render_template, request, url_for
|
||||
|
||||
|
||||
def computers_manage_icon_label(*, get_conn, clean_text, safe_next_url):
|
||||
computer_id_raw = clean_text(request.form.get("computer_id"))
|
||||
icon_label = clean_text(request.form.get("icon_label"))
|
||||
next_url = safe_next_url(request.form.get("next"), fallback="index")
|
||||
if not computer_id_raw.isdigit():
|
||||
return redirect(next_url)
|
||||
|
||||
computer_id = int(computer_id_raw)
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT 1 FROM computers WHERE id=%s", (computer_id,))
|
||||
exists = cur.fetchone() is not None
|
||||
if not exists:
|
||||
conn.close()
|
||||
return redirect(next_url)
|
||||
|
||||
if icon_label:
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO computer_manage_icons (computer_id, icon_label)
|
||||
VALUES (%s, %s)
|
||||
ON CONFLICT (computer_id)
|
||||
DO UPDATE SET icon_label = EXCLUDED.icon_label
|
||||
""",
|
||||
(computer_id, icon_label),
|
||||
)
|
||||
else:
|
||||
cur.execute("DELETE FROM computer_manage_icons WHERE computer_id=%s", (computer_id,))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
return redirect(next_url)
|
||||
|
||||
|
||||
def computers_update_cabinet(*, get_conn, log_equipment_movement):
|
||||
cid = request.form.get("id", "").strip()
|
||||
cabinet_id = request.form.get("cabinet_id", "").strip()
|
||||
if not cid.isdigit():
|
||||
flash("Некорректный компьютер")
|
||||
return redirect(request.referrer or url_for("index"))
|
||||
cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
cur.execute("SELECT inventory_number, cabinet_id FROM computers WHERE id=%s", (int(cid),))
|
||||
row = cur.fetchone()
|
||||
inv_num = row[0] if row else None
|
||||
old_cab_id = row[1] if row else None
|
||||
cur.execute("UPDATE computers SET cabinet_id=%s WHERE id=%s", (cab_id, int(cid)))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
if inv_num and old_cab_id != cab_id:
|
||||
log_equipment_movement(inv_num, "computer", old_cab_id, cab_id)
|
||||
flash("Расположение обновлено")
|
||||
return redirect(request.referrer or url_for("index"))
|
||||
|
||||
|
||||
def computers_list(
|
||||
*,
|
||||
get_conn,
|
||||
computer_types,
|
||||
computer_brands_source,
|
||||
cpu_brands,
|
||||
memory_types,
|
||||
os_options,
|
||||
):
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
cur.execute(
|
||||
"""
|
||||
SELECT c.id, c.inventory_number, c.brand, c.model, c.serial_number, c.type,
|
||||
c.cpu_brand, c.cpu_model, c.gpu_model, c.memory_size, c.memory_type,
|
||||
c.storage_size, c.motherboard, c.os, c.rustdesk_id, c.rustdesk_password,
|
||||
c.cabinet_id, cab.name AS cabinet_name,
|
||||
c.note, c.date_in_operation, c.date_added
|
||||
FROM computers c
|
||||
LEFT JOIN cabinets cab ON cab.id = c.cabinet_id
|
||||
ORDER BY
|
||||
cab.building NULLS LAST,
|
||||
CASE
|
||||
WHEN NULLIF(REGEXP_REPLACE(COALESCE(cab.cabinet, cab.name, ''), '\D', '', 'g'), '') IS NULL THEN 1
|
||||
ELSE 0
|
||||
END,
|
||||
COALESCE(NULLIF(REGEXP_REPLACE(COALESCE(cab.cabinet, cab.name, ''), '\D', '', 'g'), ''), '0')::INT,
|
||||
cab.cabinet NULLS LAST,
|
||||
cab.name NULLS LAST,
|
||||
c.inventory_number,
|
||||
c.id
|
||||
"""
|
||||
)
|
||||
rows = cur.fetchall()
|
||||
total_computers = len(rows)
|
||||
laptops_count = 0
|
||||
pcs_count = 0
|
||||
for row in rows:
|
||||
ctype = (row[5] or "").strip().lower()
|
||||
if ctype == "laptop":
|
||||
laptops_count += 1
|
||||
elif ctype == "pc":
|
||||
pcs_count += 1
|
||||
cur.execute("SELECT id, name FROM cabinets ORDER BY name")
|
||||
cabinets = cur.fetchall()
|
||||
cur.execute(computer_brands_source)
|
||||
computer_brands = [row[0] for row in cur.fetchall()]
|
||||
conn.close()
|
||||
return render_template(
|
||||
"computers.html",
|
||||
rows=rows,
|
||||
cabinets=cabinets,
|
||||
computer_types=computer_types,
|
||||
computer_brands=computer_brands,
|
||||
cpu_brands=cpu_brands,
|
||||
memory_types=memory_types,
|
||||
os_options=os_options,
|
||||
total_computers=total_computers,
|
||||
laptops_count=laptops_count,
|
||||
pcs_count=pcs_count,
|
||||
)
|
||||
|
||||
|
||||
def computers_add(*, get_conn):
|
||||
inventory_number = request.form.get("inventory_number", "").strip()
|
||||
brand = request.form.get("brand", "").strip()
|
||||
model = request.form.get("model", "").strip()
|
||||
serial_number = request.form.get("serial_number", "").strip()
|
||||
ctype = request.form.get("type", "").strip()
|
||||
cpu_brand = request.form.get("cpu_brand", "").strip()
|
||||
cpu_model = request.form.get("cpu_model", "").strip()
|
||||
gpu_model = request.form.get("gpu_model", "").strip()
|
||||
memory_size = request.form.get("memory_size", "").strip()
|
||||
memory_type = request.form.get("memory_type", "").strip()
|
||||
storage_size = request.form.get("storage_size", "").strip()
|
||||
motherboard = request.form.get("motherboard", "").strip()
|
||||
os_name = request.form.get("os", "").strip()
|
||||
rustdesk_id = request.form.get("rustdesk_id", "").strip()
|
||||
rustdesk_password = request.form.get("rustdesk_password", "").strip()
|
||||
cabinet_id = request.form.get("cabinet_id", "").strip()
|
||||
note = request.form.get("note", "").strip()
|
||||
date_in_operation = request.form.get("date_in_operation", "").strip()
|
||||
|
||||
if not inventory_number:
|
||||
flash("Укажите инвентарный номер")
|
||||
return redirect(url_for("computers"))
|
||||
if ctype not in ("pc", "laptop"):
|
||||
flash("Укажите тип (ПК/ноутбук)")
|
||||
return redirect(url_for("computers"))
|
||||
|
||||
cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
try:
|
||||
cur.execute(
|
||||
"""
|
||||
INSERT INTO computers (
|
||||
inventory_number, brand, model, serial_number, type,
|
||||
cpu_brand, cpu_model, gpu_model, memory_size, memory_type,
|
||||
storage_size, motherboard, os, rustdesk_id, rustdesk_password,
|
||||
cabinet_id, note, date_in_operation
|
||||
)
|
||||
VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)
|
||||
""",
|
||||
(
|
||||
inventory_number,
|
||||
brand or None,
|
||||
model or None,
|
||||
serial_number or None,
|
||||
ctype,
|
||||
cpu_brand or None,
|
||||
cpu_model or None,
|
||||
gpu_model or None,
|
||||
memory_size or None,
|
||||
memory_type or None,
|
||||
storage_size or None,
|
||||
motherboard or None,
|
||||
os_name or None,
|
||||
rustdesk_id or None,
|
||||
rustdesk_password or None,
|
||||
cab_id,
|
||||
note or None,
|
||||
date_in_operation or None,
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
flash("Компьютер добавлен")
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
flash("Не удалось добавить компьютер")
|
||||
finally:
|
||||
conn.close()
|
||||
return redirect(url_for("computers"))
|
||||
|
||||
|
||||
def computers_edit(*, get_conn, log_equipment_movement):
|
||||
cid = request.form.get("id", "").strip()
|
||||
inventory_number = request.form.get("inventory_number", "").strip()
|
||||
brand = request.form.get("brand", "").strip()
|
||||
model = request.form.get("model", "").strip()
|
||||
serial_number = request.form.get("serial_number", "").strip()
|
||||
ctype = request.form.get("type", "").strip()
|
||||
cpu_brand = request.form.get("cpu_brand", "").strip()
|
||||
cpu_model = request.form.get("cpu_model", "").strip()
|
||||
gpu_model = request.form.get("gpu_model", "").strip()
|
||||
memory_size = request.form.get("memory_size", "").strip()
|
||||
memory_type = request.form.get("memory_type", "").strip()
|
||||
storage_size = request.form.get("storage_size", "").strip()
|
||||
motherboard = request.form.get("motherboard", "").strip()
|
||||
os_name = request.form.get("os", "").strip()
|
||||
rustdesk_id = request.form.get("rustdesk_id", "").strip()
|
||||
rustdesk_password = request.form.get("rustdesk_password", "").strip()
|
||||
cabinet_id = request.form.get("cabinet_id", "").strip()
|
||||
note = request.form.get("note", "").strip()
|
||||
date_in_operation = request.form.get("date_in_operation", "").strip()
|
||||
|
||||
if not cid.isdigit():
|
||||
flash("Некорректное устройство")
|
||||
return redirect(url_for("computers"))
|
||||
if not inventory_number:
|
||||
flash("Укажите инвентарный номер")
|
||||
return redirect(url_for("computers"))
|
||||
if ctype not in ("pc", "laptop"):
|
||||
flash("Укажите тип устройства (ПК/ноутбук)")
|
||||
return redirect(url_for("computers"))
|
||||
|
||||
cab_id = int(cabinet_id) if cabinet_id.isdigit() else None
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
try:
|
||||
cur.execute("SELECT cabinet_id FROM computers WHERE id=%s", (int(cid),))
|
||||
row = cur.fetchone()
|
||||
old_cab_id = row[0] if row else None
|
||||
cur.execute(
|
||||
"""
|
||||
UPDATE computers
|
||||
SET inventory_number=%s, brand=%s, model=%s, serial_number=%s, type=%s,
|
||||
cpu_brand=%s, cpu_model=%s, gpu_model=%s, memory_size=%s, memory_type=%s,
|
||||
storage_size=%s, motherboard=%s, os=%s, rustdesk_id=%s, rustdesk_password=%s,
|
||||
cabinet_id=%s, note=%s, date_in_operation=%s
|
||||
WHERE id=%s
|
||||
""",
|
||||
(
|
||||
inventory_number,
|
||||
brand or None,
|
||||
model or None,
|
||||
serial_number or None,
|
||||
ctype,
|
||||
cpu_brand or None,
|
||||
cpu_model or None,
|
||||
gpu_model or None,
|
||||
memory_size or None,
|
||||
memory_type or None,
|
||||
storage_size or None,
|
||||
motherboard or None,
|
||||
os_name or None,
|
||||
rustdesk_id or None,
|
||||
rustdesk_password or None,
|
||||
cab_id,
|
||||
note or None,
|
||||
date_in_operation or None,
|
||||
int(cid),
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
if old_cab_id != cab_id:
|
||||
log_equipment_movement(inventory_number, "computer", old_cab_id, cab_id)
|
||||
flash("Компьютер обновлен")
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
flash("Не удалось обновить компьютер")
|
||||
finally:
|
||||
conn.close()
|
||||
return redirect(url_for("computers"))
|
||||
|
||||
|
||||
def computers_delete(*, get_conn):
|
||||
cid = request.form.get("id", "").strip()
|
||||
if not cid.isdigit():
|
||||
flash("Некорректное устройство")
|
||||
return redirect(url_for("computers"))
|
||||
conn = get_conn()
|
||||
cur = conn.cursor()
|
||||
cur.execute("DELETE FROM computers WHERE id=%s", (int(cid),))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
flash("Компьютер удален")
|
||||
return redirect(url_for("computers"))
|
||||
Reference in New Issue
Block a user