65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
import os
|
|
import re
|
|
from datetime import datetime, timedelta
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
DATA_DIR = os.path.join(BASE_DIR, "data")
|
|
|
|
SCUD_DIR = os.path.join(DATA_DIR, "scud")
|
|
ZUP_1C_DIR = os.path.join(DATA_DIR, "1c")
|
|
OUTPUT_DIR = os.path.join(BASE_DIR, "output")
|
|
|
|
for folder in [DATA_DIR, SCUD_DIR, ZUP_1C_DIR, OUTPUT_DIR]:
|
|
os.makedirs(folder, exist_ok=True)
|
|
|
|
NOW = datetime.now()
|
|
DATE_TODAY = NOW.strftime("%d.%m.%Y")
|
|
|
|
if NOW.weekday() == 0:
|
|
DATE_YESTERDAY = (NOW - timedelta(days=3)).strftime("%d.%m.%Y")
|
|
else:
|
|
DATE_YESTERDAY = (NOW - timedelta(days=1)).strftime("%d.%m.%Y")
|
|
|
|
SCUD_FILE_TODAY = os.path.join(SCUD_DIR, f"Сотрудники_{DATE_TODAY}.xlsx")
|
|
SCUD_FILE_YESTERDAY = os.path.join(SCUD_DIR, f"Сотрудники_{DATE_YESTERDAY}.xlsx")
|
|
SCUD_FILE = os.path.join(SCUD_DIR, "Сотрудники.xlsx")
|
|
|
|
STAFF_FILE_1C = os.path.join(ZUP_1C_DIR, "Штатные сотрудники - ОУП.xlsx")
|
|
ABSENT_FILE_1C = os.path.join(ZUP_1C_DIR, "Отсутствия сотрудников - выгрузка.xlsx")
|
|
|
|
OLLAMA_URL = "http://10.121.18.227:11434/api/generate"
|
|
OLLAMA_MODEL = "qwen2.5:14b"
|
|
MODEL_NAME = OLLAMA_MODEL
|
|
|
|
KNOWLEDGE_BASE_PATH = os.path.join(DATA_DIR, "knowledge_base.json")
|
|
EXCEPTIONS_PATH = os.path.join(BASE_DIR, "exceptions.json")
|
|
|
|
|
|
def normalize_fio(fio):
|
|
"""
|
|
Нормализует ФИО:
|
|
1. Вырезает скобки и их содержимое: (осн.), (совм.), (сводная), (внешн.)
|
|
2. Заменяет неразрывные пробелы \xa0 на обычные
|
|
3. Приводит к регистру Заглавных Букв
|
|
"""
|
|
if not fio or not isinstance(fio, str):
|
|
return ""
|
|
fio_clean = re.sub(r'\(.*?\)', '', fio)
|
|
fio_clean = fio_clean.replace('\xa0', ' ')
|
|
parts = fio_clean.strip().split()
|
|
return " ".join(parts).title()
|
|
|
|
|
|
def clean_scud_fio_light(fio_str):
|
|
return normalize_fio(fio_str)
|
|
|
|
|
|
def load_exceptions():
|
|
import json
|
|
if os.path.exists(EXCEPTIONS_PATH):
|
|
try:
|
|
with open(EXCEPTIONS_PATH, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except Exception:
|
|
pass
|
|
return {"fio": [], "departments": [], "positions": [], "position_keywords": []} |