MVP ready. Fully functional (registration? moderation, profiles vew)
Some checks reported errors
continuous-integration/drone/push Build encountered an error
Some checks reported errors
continuous-integration/drone/push Build encountered an error
This commit is contained in:
@@ -36,6 +36,9 @@ services:
|
|||||||
dockerfile: Dockerfile
|
dockerfile: Dockerfile
|
||||||
args:
|
args:
|
||||||
- PY_VERSION=3.12
|
- PY_VERSION=3.12
|
||||||
|
environment:
|
||||||
|
TZ: ${TZ}
|
||||||
|
PYTHONPATH: /app
|
||||||
depends_on:
|
depends_on:
|
||||||
db:
|
db:
|
||||||
condition: service_healthy
|
condition: service_healthy
|
||||||
|
|||||||
@@ -1,24 +1,42 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
import os
|
import os, sys
|
||||||
from logging.config import fileConfig
|
from logging.config import fileConfig
|
||||||
|
|
||||||
from sqlalchemy import engine_from_config, pool
|
from sqlalchemy import engine_from_config, pool
|
||||||
from alembic import context
|
from alembic import context
|
||||||
|
|
||||||
config = context.config
|
config = context.config
|
||||||
# Подменяем URL из переменных окружения контейнера
|
|
||||||
if os.getenv("DATABASE_URL"):
|
|
||||||
config.set_main_option("sqlalchemy.url", os.getenv("DATABASE_URL"))
|
|
||||||
|
|
||||||
fileConfig(config.config_file_name)
|
# Возьми URL из переменных контейнера
|
||||||
|
db_url = os.getenv("DATABASE_URL")
|
||||||
|
if db_url:
|
||||||
|
config.set_main_option("sqlalchemy.url", db_url)
|
||||||
|
|
||||||
|
# Лог-конфиг (без падения, если секций нет)
|
||||||
|
cfg_name = config.config_file_name
|
||||||
|
if cfg_name:
|
||||||
|
try:
|
||||||
|
fileConfig(cfg_name)
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Обеспечим PYTHONPATH=/app для импорта пакета app
|
||||||
|
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
||||||
|
if ROOT not in sys.path:
|
||||||
|
sys.path.insert(0, ROOT)
|
||||||
|
|
||||||
# Метаданные моделей
|
# Метаданные моделей
|
||||||
from app.models.base import Base # noqa: E402
|
from app.models.base import Base # noqa
|
||||||
target_metadata = Base.metadata
|
target_metadata = Base.metadata
|
||||||
|
|
||||||
def run_migrations_offline():
|
def run_migrations_offline():
|
||||||
url = config.get_main_option("sqlalchemy.url")
|
url = config.get_main_option("sqlalchemy.url")
|
||||||
context.configure(url=url, target_metadata=target_metadata, literal_binds=True,
|
context.configure(
|
||||||
dialect_opts={"paramstyle": "named"})
|
url=url,
|
||||||
|
target_metadata=target_metadata,
|
||||||
|
literal_binds=True,
|
||||||
|
dialect_opts={"paramstyle": "named"},
|
||||||
|
)
|
||||||
with context.begin_transaction():
|
with context.begin_transaction():
|
||||||
context.run_migrations()
|
context.run_migrations()
|
||||||
|
|
||||||
|
|||||||
@@ -20,6 +20,7 @@ def upgrade():
|
|||||||
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
||||||
sa.Column('telegram_id', sa.Integer(), unique=True, index=True),
|
sa.Column('telegram_id', sa.Integer(), unique=True, index=True),
|
||||||
sa.Column('username', sa.String(length=100), nullable=True),
|
sa.Column('username', sa.String(length=100), nullable=True),
|
||||||
|
|
||||||
sa.Column('full_name', sa.String(length=120), nullable=True),
|
sa.Column('full_name', sa.String(length=120), nullable=True),
|
||||||
sa.Column('gender', sa.String(length=20), nullable=True),
|
sa.Column('gender', sa.String(length=20), nullable=True),
|
||||||
sa.Column('birth_date', sa.Date(), nullable=True),
|
sa.Column('birth_date', sa.Date(), nullable=True),
|
||||||
@@ -30,6 +31,7 @@ def upgrade():
|
|||||||
sa.Column('citizenship', sa.String(length=80), nullable=True),
|
sa.Column('citizenship', sa.String(length=80), nullable=True),
|
||||||
sa.Column('visa_status', sa.String(length=60), nullable=True),
|
sa.Column('visa_status', sa.String(length=60), nullable=True),
|
||||||
sa.Column('languages', sa.String(length=200), nullable=True),
|
sa.Column('languages', sa.String(length=200), nullable=True),
|
||||||
|
|
||||||
sa.Column('education', sa.String(length=120), nullable=True),
|
sa.Column('education', sa.String(length=120), nullable=True),
|
||||||
sa.Column('occupation', sa.String(length=120), nullable=True),
|
sa.Column('occupation', sa.String(length=120), nullable=True),
|
||||||
sa.Column('income_range', sa.String(length=60), nullable=True),
|
sa.Column('income_range', sa.String(length=60), nullable=True),
|
||||||
@@ -37,19 +39,25 @@ def upgrade():
|
|||||||
sa.Column('marital_status', sa.String(length=60), nullable=True),
|
sa.Column('marital_status', sa.String(length=60), nullable=True),
|
||||||
sa.Column('has_children', sa.Boolean(), nullable=True),
|
sa.Column('has_children', sa.Boolean(), nullable=True),
|
||||||
sa.Column('children_notes', sa.String(length=200), nullable=True),
|
sa.Column('children_notes', sa.String(length=200), nullable=True),
|
||||||
|
|
||||||
sa.Column('smoking', sa.String(length=20), nullable=True),
|
sa.Column('smoking', sa.String(length=20), nullable=True),
|
||||||
sa.Column('alcohol', sa.String(length=20), nullable=True),
|
sa.Column('alcohol', sa.String(length=20), nullable=True),
|
||||||
sa.Column('health_notes', sa.String(length=300), nullable=True),
|
sa.Column('health_notes', sa.String(length=300), nullable=True),
|
||||||
|
|
||||||
sa.Column('hobbies_tags', sa.String(length=300), nullable=True),
|
sa.Column('hobbies_tags', sa.String(length=300), nullable=True),
|
||||||
sa.Column('hobbies_free', sa.Text(), nullable=True),
|
sa.Column('hobbies_free', sa.Text(), nullable=True),
|
||||||
|
|
||||||
sa.Column('goal', sa.String(length=120), nullable=True),
|
sa.Column('goal', sa.String(length=120), nullable=True),
|
||||||
sa.Column('partner_prefs', sa.Text(), nullable=True),
|
sa.Column('partner_prefs', sa.Text(), nullable=True),
|
||||||
|
|
||||||
sa.Column('avatar_file_id', sa.String(length=200), nullable=True),
|
sa.Column('avatar_file_id', sa.String(length=200), nullable=True),
|
||||||
sa.Column('gallery_file_ids', sa.Text(), nullable=True),
|
sa.Column('gallery_file_ids', sa.Text(), nullable=True),
|
||||||
|
|
||||||
sa.Column('consent_personal', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
sa.Column('consent_personal', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
||||||
sa.Column('consent_policy', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
sa.Column('consent_policy', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
||||||
sa.Column('is_verified', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
sa.Column('is_verified', sa.Boolean(), nullable=False, server_default=sa.text('0')),
|
||||||
sa.Column('is_active', sa.Boolean(), nullable=False, server_default=sa.text('1')),
|
sa.Column('is_active', sa.Boolean(), nullable=False, server_default=sa.text('1')),
|
||||||
|
|
||||||
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
||||||
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
|
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -2,4 +2,6 @@ import os
|
|||||||
class Settings:
|
class Settings:
|
||||||
BOT_TOKEN = os.getenv("BOT_TOKEN", "")
|
BOT_TOKEN = os.getenv("BOT_TOKEN", "")
|
||||||
DATABASE_URL = os.getenv("DATABASE_URL", "")
|
DATABASE_URL = os.getenv("DATABASE_URL", "")
|
||||||
|
PRIMARY_ADMIN_TELEGRAM_ID = os.getenv("PRIMARY_ADMIN_TELEGRAM_ID", "")
|
||||||
|
|
||||||
settings = Settings()
|
settings = Settings()
|
||||||
|
|||||||
22
services/bot/app/core/db.py
Normal file
22
services/bot/app/core/db.py
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from sqlalchemy import create_engine
|
||||||
|
from sqlalchemy.orm import sessionmaker, Session
|
||||||
|
from app.core.config import settings
|
||||||
|
|
||||||
|
# Создаём движок SQLAlchemy
|
||||||
|
engine = create_engine(
|
||||||
|
settings.DATABASE_URL,
|
||||||
|
future=True,
|
||||||
|
pool_pre_ping=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Фабрика сессий
|
||||||
|
SessionLocal = sessionmaker(
|
||||||
|
bind=engine,
|
||||||
|
expire_on_commit=False,
|
||||||
|
future=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
def db_session() -> Session:
|
||||||
|
"""Получить новую сессию БД."""
|
||||||
|
return SessionLocal()
|
||||||
141
services/bot/app/handlers/admin.py
Normal file
141
services/bot/app/handlers/admin.py
Normal file
@@ -0,0 +1,141 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from telegram import Update
|
||||||
|
from telegram.ext import ContextTypes
|
||||||
|
from sqlalchemy import desc
|
||||||
|
|
||||||
|
from app.core.db import db_session
|
||||||
|
from app.repositories.admin_repo import AdminRepository
|
||||||
|
from app.repositories.candidate_repo import CandidateRepository
|
||||||
|
from app.models.candidate import Candidate
|
||||||
|
|
||||||
|
from app.utils.common import csv_to_list
|
||||||
|
|
||||||
|
async def add_admin_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""Команда /addadmin <telegram_id> — добавить нового администратора (только для админов)."""
|
||||||
|
with db_session() as s:
|
||||||
|
repo = AdminRepository(s)
|
||||||
|
if not repo.is_admin(update.effective_user.id):
|
||||||
|
await update.message.reply_text("Недостаточно прав.")
|
||||||
|
return
|
||||||
|
|
||||||
|
args = context.args or []
|
||||||
|
if not args:
|
||||||
|
await update.message.reply_text("Формат: /addadmin 123456789")
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
tid = int(args[0])
|
||||||
|
except ValueError:
|
||||||
|
await update.message.reply_text("Неверный telegram_id.")
|
||||||
|
return
|
||||||
|
|
||||||
|
from app.models.admin import Admin
|
||||||
|
if s.query(Admin).filter(Admin.telegram_id == tid).first():
|
||||||
|
await update.message.reply_text("Этот администратор уже добавлен.")
|
||||||
|
return
|
||||||
|
|
||||||
|
s.add(Admin(telegram_id=tid))
|
||||||
|
s.commit()
|
||||||
|
await update.message.reply_text(f"Администратор {tid} добавлен.")
|
||||||
|
|
||||||
|
|
||||||
|
async def list_candidates_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""Команда /candidates — показать последние анкеты (только для админов)."""
|
||||||
|
with db_session() as s:
|
||||||
|
admin_repo = AdminRepository(s)
|
||||||
|
if not admin_repo.is_admin(update.effective_user.id):
|
||||||
|
await update.message.reply_text("Недостаточно прав.")
|
||||||
|
return
|
||||||
|
|
||||||
|
rows = s.query(Candidate).order_by(desc(Candidate.created_at)).limit(30).all()
|
||||||
|
if not rows:
|
||||||
|
await update.message.reply_text("Кандидатов пока нет.")
|
||||||
|
return
|
||||||
|
|
||||||
|
out = []
|
||||||
|
for c in rows:
|
||||||
|
out.append(
|
||||||
|
f"#{c.id} {c.full_name or '—'} | {c.city or '—'} | цель: {c.goal or '—'} | @{c.username or ''}"
|
||||||
|
)
|
||||||
|
await update.message.reply_text("Последние анкеты:\n" + "\n".join(out))
|
||||||
|
|
||||||
|
|
||||||
|
async def verify_candidate_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""Команда /verify <id> — пометить анкету как проверенную (только для админов)."""
|
||||||
|
with db_session() as s:
|
||||||
|
admin_repo = AdminRepository(s)
|
||||||
|
if not admin_repo.is_admin(update.effective_user.id):
|
||||||
|
await update.message.reply_text("Недостаточно прав.")
|
||||||
|
return
|
||||||
|
|
||||||
|
args = context.args or []
|
||||||
|
if not args:
|
||||||
|
await update.message.reply_text("Формат: /verify 12")
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
cid = int(args[0])
|
||||||
|
except ValueError:
|
||||||
|
await update.message.reply_text("Укажи числовой id анкеты.")
|
||||||
|
return
|
||||||
|
|
||||||
|
c = s.get(Candidate, cid)
|
||||||
|
if not c:
|
||||||
|
await update.message.reply_text("Анкета не найдена.")
|
||||||
|
return
|
||||||
|
|
||||||
|
c.is_verified = True
|
||||||
|
s.commit()
|
||||||
|
await update.message.reply_text(f"Анкета #{cid} помечена как проверенная.")
|
||||||
|
|
||||||
|
async def view_candidate_handler(update, context):
|
||||||
|
"""Команда /view <id> — показать карточку анкеты с фото (для админов)."""
|
||||||
|
with db_session() as s:
|
||||||
|
admin_repo = AdminRepository(s)
|
||||||
|
if not admin_repo.is_admin(update.effective_user.id):
|
||||||
|
await update.message.reply_text("Недостаточно прав.")
|
||||||
|
return
|
||||||
|
|
||||||
|
args = context.args or []
|
||||||
|
if not args:
|
||||||
|
await update.message.reply_text("Формат: /view 12")
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
cid = int(args[0])
|
||||||
|
except ValueError:
|
||||||
|
await update.message.reply_text("Укажи числовой id анкеты.")
|
||||||
|
return
|
||||||
|
|
||||||
|
c = s.get(Candidate, cid)
|
||||||
|
if not c:
|
||||||
|
await update.message.reply_text("Анкета не найдена.")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Текстовая карточка
|
||||||
|
caption = (
|
||||||
|
f"#{c.id} {c.full_name or '—'} @{c.username or ''}\n"
|
||||||
|
f"Город: {c.city or '—'} | Цель: {c.goal or '—'}\n"
|
||||||
|
f"Статус: {'✅ проверена' if c.is_verified else '⏳ на проверке'} | Активна: {'да' if c.is_active else 'нет'}"
|
||||||
|
)
|
||||||
|
|
||||||
|
chat_id = update.effective_chat.id
|
||||||
|
|
||||||
|
# Аватар (если есть)
|
||||||
|
if c.avatar_file_id:
|
||||||
|
await context.bot.send_photo(chat_id=chat_id, photo=c.avatar_file_id, caption=caption)
|
||||||
|
else:
|
||||||
|
await context.bot.send_message(chat_id=chat_id, text=caption + "\n(аватар отсутствует)")
|
||||||
|
|
||||||
|
# Галерея (если есть)
|
||||||
|
gallery_ids = csv_to_list(c.gallery_file_ids)
|
||||||
|
if gallery_ids:
|
||||||
|
# Telegram принимает до 10 media за раз
|
||||||
|
chunk = []
|
||||||
|
for fid in gallery_ids:
|
||||||
|
chunk.append(InputMediaPhoto(media=fid))
|
||||||
|
if len(chunk) == 10:
|
||||||
|
await context.bot.send_media_group(chat_id=chat_id, media=chunk)
|
||||||
|
chunk = []
|
||||||
|
if chunk:
|
||||||
|
await context.bot.send_media_group(chat_id=chat_id, media=chunk)
|
||||||
348
services/bot/app/handlers/conversation.py
Normal file
348
services/bot/app/handlers/conversation.py
Normal file
@@ -0,0 +1,348 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from datetime import datetime, date
|
||||||
|
from typing import List
|
||||||
|
|
||||||
|
from telegram import ReplyKeyboardMarkup, ReplyKeyboardRemove, InputMediaPhoto
|
||||||
|
from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters
|
||||||
|
|
||||||
|
from app.core.db import db_session
|
||||||
|
from app.models.candidate import Candidate
|
||||||
|
from app.repositories.admin_repo import AdminRepository
|
||||||
|
from app.utils.common import csv_to_list, list_to_csv, calc_age
|
||||||
|
from app.services.admin_bootstrap import ensure_primary_admin
|
||||||
|
|
||||||
|
(
|
||||||
|
S_NAME, S_GENDER, S_BIRTH, S_HEIGHT, S_WEIGHT,
|
||||||
|
S_COUNTRY, S_CITY, S_CITIZEN, S_VISA, S_LANGS,
|
||||||
|
S_EDU, S_OCCUP, S_INCOME, S_RELIGION, S_MARITAL, S_CHILDREN, S_CHILDREN_NOTES,
|
||||||
|
S_SMOKING, S_ALCOHOL, S_HEALTH,
|
||||||
|
S_HOBBIES_TAGS, S_HOBBIES_FREE,
|
||||||
|
S_GOAL, S_PARTNER,
|
||||||
|
S_AVATAR, S_GALLERY, S_CONSENTS, S_CONFIRM
|
||||||
|
) = range(28)
|
||||||
|
|
||||||
|
GENDER_KB = [["жен", "муж"], ["другое"]]
|
||||||
|
YESNO_KB = [["да", "нет"]]
|
||||||
|
SMOKE_ALC_KB = [["нет", "иногда", "да"]]
|
||||||
|
MARITAL_KB = [["никогда не был(а) в браке"], ["в разводе"], ["вдова/вдовец"]]
|
||||||
|
INCOME_KB = [["до 2 млн KRW", "2–4 млн"], ["4–6 млн", "6+ млн"], ["не указывать"]]
|
||||||
|
|
||||||
|
def build_conversation() -> ConversationHandler:
|
||||||
|
return ConversationHandler(
|
||||||
|
entry_points=[CommandHandler("start", start), CommandHandler("new", start)],
|
||||||
|
states={
|
||||||
|
S_NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_name)],
|
||||||
|
S_GENDER: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_gender)],
|
||||||
|
S_BIRTH: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_birth)],
|
||||||
|
S_HEIGHT: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_height)],
|
||||||
|
S_WEIGHT: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_weight)],
|
||||||
|
S_COUNTRY: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_country)],
|
||||||
|
S_CITY: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_city)],
|
||||||
|
S_CITIZEN: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_citizen)],
|
||||||
|
S_VISA: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_visa)],
|
||||||
|
S_LANGS: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_langs)],
|
||||||
|
S_EDU: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_edu)],
|
||||||
|
S_OCCUP: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_occup)],
|
||||||
|
S_INCOME: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_income)],
|
||||||
|
S_RELIGION: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_religion)],
|
||||||
|
S_MARITAL: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_marital)],
|
||||||
|
S_CHILDREN: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_children)],
|
||||||
|
S_CHILDREN_NOTES: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_children_notes)],
|
||||||
|
S_SMOKING: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_smoking)],
|
||||||
|
S_ALCOHOL: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_alcohol)],
|
||||||
|
S_HEALTH: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_health)],
|
||||||
|
S_HOBBIES_TAGS: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_hobbies_tags)],
|
||||||
|
S_HOBBIES_FREE: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_hobbies_free)],
|
||||||
|
S_GOAL: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_goal)],
|
||||||
|
S_PARTNER: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_partner)],
|
||||||
|
S_AVATAR: [
|
||||||
|
MessageHandler(filters.PHOTO, s_avatar),
|
||||||
|
MessageHandler(filters.TEXT & ~filters.COMMAND, s_avatar),
|
||||||
|
],
|
||||||
|
S_GALLERY: [
|
||||||
|
MessageHandler(filters.PHOTO, s_gallery),
|
||||||
|
MessageHandler(filters.TEXT & ~filters.COMMAND, s_gallery),
|
||||||
|
],
|
||||||
|
S_CONSENTS: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_consents)],
|
||||||
|
S_CONFIRM: [MessageHandler(filters.TEXT & ~filters.COMMAND, s_confirm)],
|
||||||
|
},
|
||||||
|
fallbacks=[CommandHandler("cancel", cancel)],
|
||||||
|
allow_reentry=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def start(update, context):
|
||||||
|
with db_session() as s:
|
||||||
|
ensure_primary_admin(s)
|
||||||
|
exists = s.query(Candidate).filter(Candidate.telegram_id == update.effective_user.id).first()
|
||||||
|
if exists:
|
||||||
|
await update.message.reply_text(
|
||||||
|
"Анкета уже существует. Посмотреть — /my, редактирование — /edit (скоро), начать заново — /new",
|
||||||
|
reply_markup=ReplyKeyboardRemove(),
|
||||||
|
)
|
||||||
|
return ConversationHandler.END
|
||||||
|
|
||||||
|
await update.message.reply_text(
|
||||||
|
"Привет! Это анкета для брачного сервиса с сопровождением до визы F‑6 🇰🇷\nКак тебя зовут?",
|
||||||
|
reply_markup=ReplyKeyboardRemove(),
|
||||||
|
)
|
||||||
|
return S_NAME
|
||||||
|
|
||||||
|
async def s_name(update, context):
|
||||||
|
context.user_data["full_name"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Твой пол?", reply_markup=ReplyKeyboardMarkup(GENDER_KB, one_time_keyboard=True, resize_keyboard=True))
|
||||||
|
return S_GENDER
|
||||||
|
|
||||||
|
async def s_gender(update, context):
|
||||||
|
context.user_data["gender"] = update.message.text.strip().lower()
|
||||||
|
await update.message.reply_text("Дата рождения (ДД.ММ.ГГГГ):", reply_markup=ReplyKeyboardRemove())
|
||||||
|
return S_BIRTH
|
||||||
|
|
||||||
|
async def s_birth(update, context):
|
||||||
|
try:
|
||||||
|
d = datetime.strptime(update.message.text.strip(), "%d.%m.%Y").date()
|
||||||
|
except Exception:
|
||||||
|
await update.message.reply_text("Неверный формат. Пример: 31.12.1995")
|
||||||
|
return S_BIRTH
|
||||||
|
context.user_data["birth_date"] = d
|
||||||
|
await update.message.reply_text("Рост (см):")
|
||||||
|
return S_HEIGHT
|
||||||
|
|
||||||
|
async def s_height(update, context):
|
||||||
|
try:
|
||||||
|
context.user_data["height_cm"] = float(update.message.text.replace(",", "."))
|
||||||
|
except Exception:
|
||||||
|
await update.message.reply_text("Введи число, например 168")
|
||||||
|
return S_HEIGHT
|
||||||
|
await update.message.reply_text("Вес (кг):")
|
||||||
|
return S_WEIGHT
|
||||||
|
|
||||||
|
async def s_weight(update, context):
|
||||||
|
try:
|
||||||
|
context.user_data["weight_kg"] = float(update.message.text.replace(",", "."))
|
||||||
|
except Exception:
|
||||||
|
await update.message.reply_text("Введи число, например 55")
|
||||||
|
return S_WEIGHT
|
||||||
|
await update.message.reply_text("Страна проживания:")
|
||||||
|
return S_COUNTRY
|
||||||
|
|
||||||
|
async def s_country(update, context):
|
||||||
|
context.user_data["country"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Город:")
|
||||||
|
return S_CITY
|
||||||
|
|
||||||
|
async def s_city(update, context):
|
||||||
|
context.user_data["city"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Гражданство (РФ/Казахстан/Узбекистан и т.п.):")
|
||||||
|
return S_CITIZEN
|
||||||
|
|
||||||
|
async def s_citizen(update, context):
|
||||||
|
context.user_data["citizenship"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Текущий визовый статус в Корее (нет/F‑4/H‑1 и т.п.):")
|
||||||
|
return S_VISA
|
||||||
|
|
||||||
|
async def s_visa(update, context):
|
||||||
|
context.user_data["visa_status"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Языки (через запятую, напр.: ru, ko, en):")
|
||||||
|
return S_LANGS
|
||||||
|
|
||||||
|
async def s_langs(update, context):
|
||||||
|
context.user_data["languages"] = list_to_csv(csv_to_list(update.message.text))
|
||||||
|
await update.message.reply_text("Образование (высшее/колледж/иное):")
|
||||||
|
return S_EDU
|
||||||
|
|
||||||
|
async def s_edu(update, context):
|
||||||
|
context.user_data["education"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Профессия:")
|
||||||
|
return S_OCCUP
|
||||||
|
|
||||||
|
async def s_occup(update, context):
|
||||||
|
context.user_data["occupation"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text(
|
||||||
|
"Доход (диапазон):",
|
||||||
|
reply_markup=ReplyKeyboardMarkup(INCOME_KB, one_time_keyboard=True, resize_keyboard=True),
|
||||||
|
)
|
||||||
|
return S_INCOME
|
||||||
|
|
||||||
|
async def s_income(update, context):
|
||||||
|
context.user_data["income_range"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Религия (если нет, напиши «нет»):", reply_markup=ReplyKeyboardRemove())
|
||||||
|
return S_RELIGION
|
||||||
|
|
||||||
|
async def s_religion(update, context):
|
||||||
|
context.user_data["religion"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text(
|
||||||
|
"Семейное положение:",
|
||||||
|
reply_markup=ReplyKeyboardMarkup(MARITAL_KB, one_time_keyboard=True, resize_keyboard=True),
|
||||||
|
)
|
||||||
|
return S_MARITAL
|
||||||
|
|
||||||
|
async def s_marital(update, context):
|
||||||
|
context.user_data["marital_status"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Есть дети?", reply_markup=ReplyKeyboardMarkup(YESNO_KB, one_time_keyboard=True, resize_keyboard=True))
|
||||||
|
return S_CHILDREN
|
||||||
|
|
||||||
|
async def s_children(update, context):
|
||||||
|
ans = update.message.text.strip().lower()
|
||||||
|
context.user_data["has_children"] = (ans == "да")
|
||||||
|
if ans == "да":
|
||||||
|
await update.message.reply_text("Коротко про детей (кол-во/возраст/с кем проживают):", reply_markup=ReplyKeyboardRemove())
|
||||||
|
return S_CHILDREN_NOTES
|
||||||
|
context.user_data["children_notes"] = None
|
||||||
|
await update.message.reply_text("Курение:", reply_markup=ReplyKeyboardMarkup(SMOKE_ALC_KB, one_time_keyboard=True, resize_keyboard=True))
|
||||||
|
return S_SMOKING
|
||||||
|
|
||||||
|
async def s_children_notes(update, context):
|
||||||
|
context.user_data["children_notes"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Курение:", reply_markup=ReplyKeyboardMarkup(SMOKE_ALC_KB, one_time_keyboard=True, resize_keyboard=True))
|
||||||
|
return S_SMOKING
|
||||||
|
|
||||||
|
async def s_smoking(update, context):
|
||||||
|
context.user_data["smoking"] = update.message.text.strip().lower()
|
||||||
|
await update.message.reply_text("Алкоголь:", reply_markup=ReplyKeyboardMarkup(SMOKE_ALC_KB, one_time_keyboard=True, resize_keyboard=True))
|
||||||
|
return S_ALCOHOL
|
||||||
|
|
||||||
|
async def s_alcohol(update, context):
|
||||||
|
context.user_data["alcohol"] = update.message.text.strip().lower()
|
||||||
|
await update.message.reply_text("Здоровье/особенности (если есть; иначе «нет»):", reply_markup=ReplyKeyboardRemove())
|
||||||
|
return S_HEALTH
|
||||||
|
|
||||||
|
async def s_health(update, context):
|
||||||
|
context.user_data["health_notes"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Хобби/интересы (теги через запятую, напр.: пение, путешествия, k‑drama):")
|
||||||
|
return S_HOBBIES_TAGS
|
||||||
|
|
||||||
|
async def s_hobbies_tags(update, context):
|
||||||
|
context.user_data["hobbies_tags"] = list_to_csv(csv_to_list(update.message.text))
|
||||||
|
await update.message.reply_text("Расскажи подробнее про интересы (свободный текст):")
|
||||||
|
return S_HOBBIES_FREE
|
||||||
|
|
||||||
|
async def s_hobbies_free(update, context):
|
||||||
|
context.user_data["hobbies_free"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Твоя основная цель (например: «брак с гражданином Кореи (F‑6)»):")
|
||||||
|
return S_GOAL
|
||||||
|
|
||||||
|
async def s_goal(update, context):
|
||||||
|
context.user_data["goal"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Пожелания к партнёру (возраст, ценности, город, религия и т.п.):")
|
||||||
|
return S_PARTNER
|
||||||
|
|
||||||
|
async def s_partner(update, context):
|
||||||
|
context.user_data["partner_prefs"] = update.message.text.strip()
|
||||||
|
await update.message.reply_text("Загрузи своё фото (аватар):")
|
||||||
|
return S_AVATAR
|
||||||
|
|
||||||
|
async def s_avatar(update, context):
|
||||||
|
if not update.message.photo:
|
||||||
|
await update.message.reply_text("Пожалуйста, пришли фотографию.")
|
||||||
|
return S_AVATAR
|
||||||
|
context.user_data["avatar_file_id"] = update.message.photo[-1].file_id
|
||||||
|
context.user_data["gallery_file_ids"] = []
|
||||||
|
await update.message.reply_text("Можно добавить до 9 дополнительных фото. Пришли их. Когда закончишь — напиши «готово».")
|
||||||
|
return S_GALLERY
|
||||||
|
|
||||||
|
async def s_gallery(update, context):
|
||||||
|
if update.message.photo:
|
||||||
|
fid = update.message.photo[-1].file_id
|
||||||
|
gallery: List[str] = context.user_data.get("gallery_file_ids", [])
|
||||||
|
if len(gallery) < 9:
|
||||||
|
gallery.append(fid)
|
||||||
|
context.user_data["gallery_file_ids"] = gallery
|
||||||
|
await update.message.reply_text(f"Добавлено фото ({len(gallery)}/9). Ещё? Или напиши «готово».")
|
||||||
|
else:
|
||||||
|
await update.message.reply_text("Лимит 9 фото достигнут. Напиши «готово».")
|
||||||
|
return S_GALLERY
|
||||||
|
|
||||||
|
if (update.message.text or "").strip().lower() != "готово":
|
||||||
|
await update.message.reply_text("Если с фото закончили — напиши «готово».")
|
||||||
|
return S_GALLERY
|
||||||
|
|
||||||
|
await update.message.reply_text("Подтверди согласие на обработку персональных данных. Напиши: «согласен» или «согласна».")
|
||||||
|
return S_CONSENTS
|
||||||
|
|
||||||
|
async def s_consents(update, context):
|
||||||
|
ans = (update.message.text or "").strip().lower()
|
||||||
|
if ans not in ("согласен", "согласна"):
|
||||||
|
await update.message.reply_text("Нужно написать «согласен» или «согласна».")
|
||||||
|
return S_CONSENTS
|
||||||
|
context.user_data["consent_personal"] = True
|
||||||
|
context.user_data["consent_policy"] = True
|
||||||
|
|
||||||
|
d = context.user_data
|
||||||
|
age = calc_age(d.get("birth_date"))
|
||||||
|
summary = (
|
||||||
|
f"Проверь данные:\n\n"
|
||||||
|
f"Имя: {d.get('full_name')}\n"
|
||||||
|
f"Пол: {d.get('gender')}\n"
|
||||||
|
f"Дата рождения: {d.get('birth_date')} (возраст: {age})\n"
|
||||||
|
f"Рост/вес: {d.get('height_cm')} см / {d.get('weight_kg')} кг\n"
|
||||||
|
f"Страна/город: {d.get('country')}, {d.get('city')}\n"
|
||||||
|
f"Гражданство: {d.get('citizenship')}\n"
|
||||||
|
f"Виза/статус: {d.get('visa_status')}\n"
|
||||||
|
f"Языки: {d.get('languages')}\n"
|
||||||
|
f"Образование/Профессия: {d.get('education')} / {d.get('occupation')}\n"
|
||||||
|
f"Доход: {d.get('income_range')}\n"
|
||||||
|
f"Религия: {d.get('religion')}\n"
|
||||||
|
f"Семейное положение: {d.get('marital_status')}\n"
|
||||||
|
f"Дети: {'да' if d.get('has_children') else 'нет'} {'(' + d.get('children_notes') + ')' if d.get('children_notes') else ''}\n"
|
||||||
|
f"Курение/Алкоголь: {d.get('smoking')} / {d.get('alcohol')}\n"
|
||||||
|
f"Здоровье: {d.get('health_notes')}\n"
|
||||||
|
f"Хобби (теги): {d.get('hobbies_tags')}\n"
|
||||||
|
f"Интересы (текст): {d.get('hobbies_free')}\n"
|
||||||
|
f"Цель: {d.get('goal')}\n"
|
||||||
|
f"Пожелания к партнёру: {d.get('partner_prefs')}\n"
|
||||||
|
f"\nЕсли всё ок — напиши «подтверждаю». Иначе /cancel и /new."
|
||||||
|
)
|
||||||
|
await update.message.reply_text(summary)
|
||||||
|
return S_CONFIRM
|
||||||
|
|
||||||
|
async def s_confirm(update, context):
|
||||||
|
if (update.message.text or "").strip().lower() != "подтверждаю":
|
||||||
|
await update.message.reply_text('Напиши «подтверждаю» для сохранения, либо /cancel.')
|
||||||
|
return S_CONFIRM
|
||||||
|
|
||||||
|
u = update.effective_user
|
||||||
|
d = context.user_data
|
||||||
|
with db_session() as s:
|
||||||
|
cand = s.query(Candidate).filter(Candidate.telegram_id == u.id).first()
|
||||||
|
if not cand:
|
||||||
|
cand = Candidate(telegram_id=u.id, username=u.username or None)
|
||||||
|
s.add(cand)
|
||||||
|
|
||||||
|
cand.full_name = d.get("full_name")
|
||||||
|
cand.gender = d.get("gender")
|
||||||
|
cand.birth_date = d.get("birth_date")
|
||||||
|
cand.height_cm = d.get("height_cm")
|
||||||
|
cand.weight_kg = d.get("weight_kg")
|
||||||
|
cand.country = d.get("country")
|
||||||
|
cand.city = d.get("city")
|
||||||
|
cand.citizenship = d.get("citizenship")
|
||||||
|
cand.visa_status = d.get("visa_status")
|
||||||
|
cand.languages = d.get("languages")
|
||||||
|
cand.education = d.get("education")
|
||||||
|
cand.occupation = d.get("occupation")
|
||||||
|
cand.income_range = d.get("income_range")
|
||||||
|
cand.religion = d.get("religion")
|
||||||
|
cand.marital_status = d.get("marital_status")
|
||||||
|
cand.has_children = d.get("has_children")
|
||||||
|
cand.children_notes = d.get("children_notes")
|
||||||
|
cand.smoking = d.get("smoking")
|
||||||
|
cand.alcohol = d.get("alcohol")
|
||||||
|
cand.health_notes = d.get("health_notes")
|
||||||
|
cand.hobbies_tags = d.get("hobbies_tags")
|
||||||
|
cand.hobbies_free = d.get("hobbies_free")
|
||||||
|
cand.goal = d.get("goal")
|
||||||
|
cand.partner_prefs = d.get("partner_prefs")
|
||||||
|
cand.avatar_file_id = d.get("avatar_file_id")
|
||||||
|
cand.gallery_file_ids = list_to_csv(d.get("gallery_file_ids", []))
|
||||||
|
cand.consent_personal = d.get("consent_personal", False)
|
||||||
|
cand.consent_policy = d.get("consent_policy", False)
|
||||||
|
cand.is_verified = False
|
||||||
|
cand.is_active = True
|
||||||
|
s.commit()
|
||||||
|
|
||||||
|
await update.message.reply_text("Готово! Анкета сохранена. Свяжемся после модерации. Спасибо!")
|
||||||
|
return ConversationHandler.END
|
||||||
|
|
||||||
|
async def cancel(update, context):
|
||||||
|
await update.message.reply_text("Ок, отменил. Для нового старта — /new")
|
||||||
|
return ConversationHandler.END
|
||||||
41
services/bot/app/handlers/profile.py
Normal file
41
services/bot/app/handlers/profile.py
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from telegram import Update, InputMediaPhoto
|
||||||
|
from telegram.ext import ContextTypes
|
||||||
|
|
||||||
|
from app.core.db import db_session
|
||||||
|
from app.repositories.candidate_repo import CandidateRepository
|
||||||
|
from app.utils.common import csv_to_list
|
||||||
|
|
||||||
|
|
||||||
|
async def my_profile_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""Команда /my — показать свою анкету."""
|
||||||
|
with db_session() as s:
|
||||||
|
repo = CandidateRepository(s)
|
||||||
|
c = repo.by_tg(update.effective_user.id)
|
||||||
|
if not c:
|
||||||
|
await update.message.reply_text("Анкета не найдена. Начни с /new")
|
||||||
|
return
|
||||||
|
|
||||||
|
text = (
|
||||||
|
f"Твоя анкета:\n"
|
||||||
|
f"{c.full_name or '—'} (@{update.effective_user.username or ''})\n"
|
||||||
|
f"Город: {c.city or '—'}\n"
|
||||||
|
f"Цель: {c.goal or '—'}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if c.avatar_file_id:
|
||||||
|
await update.message.reply_photo(photo=c.avatar_file_id, caption=text)
|
||||||
|
else:
|
||||||
|
await update.message.reply_text(text)
|
||||||
|
|
||||||
|
gallery = csv_to_list(c.gallery_file_ids)
|
||||||
|
if gallery:
|
||||||
|
media = [InputMediaPhoto(g) for g in gallery[:10]]
|
||||||
|
await update.message.reply_media_group(media)
|
||||||
|
|
||||||
|
|
||||||
|
async def edit_hint_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""Команда /edit — пока только подсказка о редактировании."""
|
||||||
|
await update.message.reply_text(
|
||||||
|
"Точечное редактирование будет добавлено командами /edit_*. Пока что для правок — /new (перезаполнить анкету)."
|
||||||
|
)
|
||||||
@@ -1,14 +1,28 @@
|
|||||||
import asyncio
|
from __future__ import annotations
|
||||||
from telegram.ext import Application, CommandHandler
|
from telegram.ext import Application, CommandHandler
|
||||||
from app.core.config import settings
|
from app.core.config import settings
|
||||||
async def start(update, context):
|
from app.handlers.conversation import build_conversation
|
||||||
await update.message.reply_text("Привет! Бот запущен.")
|
from app.handlers.admin import add_admin_handler, list_candidates_handler, verify_candidate_handler, view_candidate_handler
|
||||||
async def main():
|
|
||||||
|
from app.handlers.profile import my_profile_handler, edit_hint_handler
|
||||||
|
|
||||||
|
async def start_cmd(update, context):
|
||||||
|
await update.message.reply_text("Бот запущен. Набери /new, чтобы заполнить анкету.")
|
||||||
|
|
||||||
|
def build_app() -> Application:
|
||||||
app = Application.builder().token(settings.BOT_TOKEN).build()
|
app = Application.builder().token(settings.BOT_TOKEN).build()
|
||||||
app.add_handler(CommandHandler("start", start))
|
# Хендлеры
|
||||||
await app.initialize()
|
app.add_handler(build_conversation())
|
||||||
await app.start()
|
app.add_handler(CommandHandler("start", start_cmd))
|
||||||
await app.updater.start_polling()
|
app.add_handler(CommandHandler("my", my_profile_handler))
|
||||||
await app.updater.wait_until_closed()
|
app.add_handler(CommandHandler("edit", edit_hint_handler))
|
||||||
|
app.add_handler(CommandHandler("addadmin", add_admin_handler))
|
||||||
|
app.add_handler(CommandHandler("candidates", list_candidates_handler))
|
||||||
|
app.add_handler(CommandHandler("verify", verify_candidate_handler))
|
||||||
|
app.add_handler(CommandHandler("view", view_candidate_handler))
|
||||||
|
return app
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
asyncio.run(main())
|
application = build_app()
|
||||||
|
# В v21 это блокирующий вызов; updater больше не используется.
|
||||||
|
application.run_polling(close_loop=False)
|
||||||
|
|||||||
3
services/bot/app/models/__init__.py
Normal file
3
services/bot/app/models/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
from .base import Base # noqa
|
||||||
|
from .admin import Admin # noqa
|
||||||
|
from .candidate import Candidate # noqa
|
||||||
14
services/bot/app/models/admin.py
Normal file
14
services/bot/app/models/admin.py
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from datetime import datetime
|
||||||
|
from sqlalchemy import Integer, String, DateTime
|
||||||
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
from app.models.base import Base
|
||||||
|
|
||||||
|
class Admin(Base):
|
||||||
|
"""Модель администратора бота."""
|
||||||
|
__tablename__ = "admins"
|
||||||
|
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
telegram_id: Mapped[int] = mapped_column(Integer, unique=True, index=True)
|
||||||
|
full_name: Mapped[str | None] = mapped_column(String(120), default=None)
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||||
6
services/bot/app/models/base.py
Normal file
6
services/bot/app/models/base.py
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from sqlalchemy.orm import DeclarativeBase
|
||||||
|
|
||||||
|
class Base(DeclarativeBase):
|
||||||
|
"""Базовый класс для всех моделей."""
|
||||||
|
pass
|
||||||
53
services/bot/app/models/candidate.py
Normal file
53
services/bot/app/models/candidate.py
Normal file
@@ -0,0 +1,53 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from datetime import date, datetime
|
||||||
|
from sqlalchemy import Integer, String, Date, DateTime, Boolean, Float, Text
|
||||||
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
from app.models.base import Base
|
||||||
|
|
||||||
|
class Candidate(Base):
|
||||||
|
"""Модель анкеты кандидата."""
|
||||||
|
__tablename__ = "candidates"
|
||||||
|
|
||||||
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
telegram_id: Mapped[int] = mapped_column(Integer, unique=True, index=True)
|
||||||
|
username: Mapped[str | None] = mapped_column(String(100), default=None)
|
||||||
|
|
||||||
|
full_name: Mapped[str | None] = mapped_column(String(120))
|
||||||
|
gender: Mapped[str | None] = mapped_column(String(20))
|
||||||
|
birth_date: Mapped[date | None] = mapped_column(Date)
|
||||||
|
height_cm: Mapped[float | None] = mapped_column(Float)
|
||||||
|
weight_kg: Mapped[float | None] = mapped_column(Float)
|
||||||
|
country: Mapped[str | None] = mapped_column(String(80))
|
||||||
|
city: Mapped[str | None] = mapped_column(String(120))
|
||||||
|
citizenship: Mapped[str | None] = mapped_column(String(80))
|
||||||
|
visa_status: Mapped[str | None] = mapped_column(String(60))
|
||||||
|
languages: Mapped[str | None] = mapped_column(String(200))
|
||||||
|
|
||||||
|
education: Mapped[str | None] = mapped_column(String(120))
|
||||||
|
occupation: Mapped[str | None] = mapped_column(String(120))
|
||||||
|
income_range: Mapped[str | None] = mapped_column(String(60))
|
||||||
|
religion: Mapped[str | None] = mapped_column(String(80))
|
||||||
|
marital_status: Mapped[str | None] = mapped_column(String(60))
|
||||||
|
has_children: Mapped[bool | None] = mapped_column(Boolean, default=None)
|
||||||
|
children_notes: Mapped[str | None] = mapped_column(String(200))
|
||||||
|
|
||||||
|
smoking: Mapped[str | None] = mapped_column(String(20))
|
||||||
|
alcohol: Mapped[str | None] = mapped_column(String(20))
|
||||||
|
health_notes: Mapped[str | None] = mapped_column(String(300))
|
||||||
|
|
||||||
|
hobbies_tags: Mapped[str | None] = mapped_column(String(300))
|
||||||
|
hobbies_free: Mapped[str | None] = mapped_column(Text)
|
||||||
|
|
||||||
|
goal: Mapped[str | None] = mapped_column(String(120))
|
||||||
|
partner_prefs: Mapped[str | None] = mapped_column(Text)
|
||||||
|
|
||||||
|
avatar_file_id: Mapped[str | None] = mapped_column(String(200))
|
||||||
|
gallery_file_ids: Mapped[str | None] = mapped_column(Text)
|
||||||
|
|
||||||
|
consent_personal: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||||
|
consent_policy: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||||
|
is_verified: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||||
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||||
|
|
||||||
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||||
24
services/bot/app/repositories/admin_repo.py
Normal file
24
services/bot/app/repositories/admin_repo.py
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from app.models.admin import Admin
|
||||||
|
|
||||||
|
class AdminRepository:
|
||||||
|
"""Работа с таблицей администраторов."""
|
||||||
|
|
||||||
|
def __init__(self, session: Session):
|
||||||
|
self.s = session
|
||||||
|
|
||||||
|
def is_admin(self, telegram_id: int) -> bool:
|
||||||
|
"""Проверка, является ли пользователь администратором."""
|
||||||
|
return self.s.query(Admin).filter(Admin.telegram_id == telegram_id).first() is not None
|
||||||
|
|
||||||
|
def add_admin(self, telegram_id: int, full_name: str | None = None) -> Admin:
|
||||||
|
"""Добавить нового администратора."""
|
||||||
|
admin = Admin(telegram_id=telegram_id, full_name=full_name)
|
||||||
|
self.s.add(admin)
|
||||||
|
self.s.commit()
|
||||||
|
return admin
|
||||||
|
|
||||||
|
def get_all_admins(self) -> list[Admin]:
|
||||||
|
"""Получить всех администраторов."""
|
||||||
|
return self.s.query(Admin).all()
|
||||||
40
services/bot/app/repositories/candidate_repo.py
Normal file
40
services/bot/app/repositories/candidate_repo.py
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from app.models.candidate import Candidate
|
||||||
|
|
||||||
|
|
||||||
|
class CandidateRepository:
|
||||||
|
"""Работа с таблицей кандидатов."""
|
||||||
|
|
||||||
|
def __init__(self, session: Session):
|
||||||
|
self.s = session
|
||||||
|
|
||||||
|
def by_tg(self, telegram_id: int) -> Candidate | None:
|
||||||
|
"""Найти анкету по Telegram ID."""
|
||||||
|
return self.s.query(Candidate).filter(Candidate.telegram_id == telegram_id).first()
|
||||||
|
|
||||||
|
def by_id(self, candidate_id: int) -> Candidate | None:
|
||||||
|
"""Найти анкету по ID."""
|
||||||
|
return self.s.get(Candidate, candidate_id)
|
||||||
|
|
||||||
|
def all(self, limit: int = 100) -> list[Candidate]:
|
||||||
|
"""Получить список всех анкет (с лимитом)."""
|
||||||
|
return self.s.query(Candidate).order_by(Candidate.created_at.desc()).limit(limit).all()
|
||||||
|
|
||||||
|
def add(self, candidate: Candidate) -> Candidate:
|
||||||
|
"""Добавить новую анкету."""
|
||||||
|
self.s.add(candidate)
|
||||||
|
self.s.commit()
|
||||||
|
self.s.refresh(candidate)
|
||||||
|
return candidate
|
||||||
|
|
||||||
|
def update(self, candidate: Candidate) -> Candidate:
|
||||||
|
"""Обновить существующую анкету."""
|
||||||
|
self.s.commit()
|
||||||
|
self.s.refresh(candidate)
|
||||||
|
return candidate
|
||||||
|
|
||||||
|
def delete(self, candidate: Candidate) -> None:
|
||||||
|
"""Удалить анкету."""
|
||||||
|
self.s.delete(candidate)
|
||||||
|
self.s.commit()
|
||||||
12
services/bot/app/services/admin_bootstrap.py
Normal file
12
services/bot/app/services/admin_bootstrap.py
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from datetime import datetime
|
||||||
|
from sqlalchemy.orm import Session
|
||||||
|
from app.core.config import settings
|
||||||
|
from app.models.admin import Admin
|
||||||
|
|
||||||
|
def ensure_primary_admin(session: Session) -> None:
|
||||||
|
if settings.PRIMARY_ADMIN_TELEGRAM_ID and not session.query(Admin).filter(
|
||||||
|
Admin.telegram_id == settings.PRIMARY_ADMIN_TELEGRAM_ID
|
||||||
|
).first():
|
||||||
|
session.add(Admin(telegram_id=settings.PRIMARY_ADMIN_TELEGRAM_ID, full_name="Primary Admin", created_at=datetime.utcnow()))
|
||||||
|
session.commit()
|
||||||
15
services/bot/app/utils/common.py
Normal file
15
services/bot/app/utils/common.py
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
from datetime import date
|
||||||
|
from typing import List, Optional
|
||||||
|
|
||||||
|
def csv_to_list(s: Optional[str]) -> List[str]:
|
||||||
|
return [x.strip() for x in (s or "").split(",") if x.strip()]
|
||||||
|
|
||||||
|
def list_to_csv(items: List[str]) -> str:
|
||||||
|
return ", ".join([i.strip() for i in items if i.strip()])
|
||||||
|
|
||||||
|
def calc_age(born: Optional[date]) -> Optional[int]:
|
||||||
|
if not born:
|
||||||
|
return None
|
||||||
|
today = date.today()
|
||||||
|
return today.year - born.year - ((today.month, today.day) < (born.month, born.day))
|
||||||
Reference in New Issue
Block a user