init
This commit is contained in:
177
.gitignore
vendored
Normal file
177
.gitignore
vendored
Normal file
@@ -0,0 +1,177 @@
|
|||||||
|
# Python
|
||||||
|
__pycache__/
|
||||||
|
*.py[cod]
|
||||||
|
*$py.class
|
||||||
|
*.so
|
||||||
|
.Python
|
||||||
|
build/
|
||||||
|
develop-eggs/
|
||||||
|
dist/
|
||||||
|
downloads/
|
||||||
|
eggs/
|
||||||
|
.eggs/
|
||||||
|
lib/
|
||||||
|
lib64/
|
||||||
|
parts/
|
||||||
|
sbtarget/
|
||||||
|
var/
|
||||||
|
wheels/
|
||||||
|
share/python-wheels/
|
||||||
|
*.egg-info/
|
||||||
|
.installed.cfg
|
||||||
|
*.egg
|
||||||
|
MANIFEST
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# Installer logs
|
||||||
|
pip-log.txt
|
||||||
|
pip-delete-this-directory.txt
|
||||||
|
|
||||||
|
# Unit test / coverage reports
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
.cache
|
||||||
|
nosetests.xml
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
*.py,cover
|
||||||
|
.hypothesis/
|
||||||
|
.pytest_cache/
|
||||||
|
cover/
|
||||||
|
|
||||||
|
# Translations
|
||||||
|
*.mo
|
||||||
|
*.pot
|
||||||
|
|
||||||
|
# Django stuff:
|
||||||
|
*.log
|
||||||
|
local_settings.py
|
||||||
|
db.sqlite3
|
||||||
|
db.sqlite3-journal
|
||||||
|
|
||||||
|
# Flask stuff:
|
||||||
|
instance/
|
||||||
|
.webassets-cache
|
||||||
|
|
||||||
|
# Scrapy stuff:
|
||||||
|
.scrapy
|
||||||
|
|
||||||
|
# Sphinx documentation
|
||||||
|
docs/_build/
|
||||||
|
|
||||||
|
# PyBuilder
|
||||||
|
.pybuilder/
|
||||||
|
target/
|
||||||
|
|
||||||
|
# Jupyter Notebook
|
||||||
|
.ipynb_checkpoints
|
||||||
|
|
||||||
|
# IPython
|
||||||
|
profile_default/
|
||||||
|
ipython_config.py
|
||||||
|
|
||||||
|
# pyenv
|
||||||
|
.python-version
|
||||||
|
|
||||||
|
# pipenv
|
||||||
|
Pipfile.lock
|
||||||
|
|
||||||
|
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
|
||||||
|
__pypackages__/
|
||||||
|
|
||||||
|
# Celery stuff
|
||||||
|
celerybeat-schedule
|
||||||
|
celerybeat.pid
|
||||||
|
|
||||||
|
# SageMath parsed files
|
||||||
|
*.sage.py
|
||||||
|
|
||||||
|
# Environments
|
||||||
|
.env
|
||||||
|
.venv
|
||||||
|
env/
|
||||||
|
venv/
|
||||||
|
ENV/
|
||||||
|
env.bak/
|
||||||
|
venv.bak/
|
||||||
|
|
||||||
|
# Spyder project settings
|
||||||
|
.spyderproject
|
||||||
|
.spyproject
|
||||||
|
|
||||||
|
# Rope project settings
|
||||||
|
.ropeproject
|
||||||
|
|
||||||
|
# mkdocs documentation
|
||||||
|
/site
|
||||||
|
|
||||||
|
# mypy
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
dmypy.json
|
||||||
|
|
||||||
|
# Pyre type checker
|
||||||
|
.pyre/
|
||||||
|
|
||||||
|
# pytype static type analyzer
|
||||||
|
.pytype/
|
||||||
|
|
||||||
|
# Cython debug symbols
|
||||||
|
cython_debug/
|
||||||
|
|
||||||
|
# Медиа файлы
|
||||||
|
*.mp3
|
||||||
|
*.mp4
|
||||||
|
*.mp4.info.json
|
||||||
|
*.mp3.info.json
|
||||||
|
*.jpg
|
||||||
|
*.jpeg
|
||||||
|
*.png
|
||||||
|
*.gif
|
||||||
|
*.mov
|
||||||
|
*.avi
|
||||||
|
*.mkv
|
||||||
|
*.webm
|
||||||
|
*.flv
|
||||||
|
|
||||||
|
# Кэш и временные файлы бота
|
||||||
|
/cache/
|
||||||
|
/tmp/
|
||||||
|
downloads/
|
||||||
|
*.cache
|
||||||
|
*.tmp
|
||||||
|
|
||||||
|
# Конфигурационные файлы с чувствительными данными
|
||||||
|
config.py
|
||||||
|
secrets.py
|
||||||
|
cookies.txt
|
||||||
|
*.key
|
||||||
|
*.pem
|
||||||
|
*.crt
|
||||||
|
|
||||||
|
# Логи
|
||||||
|
*.log
|
||||||
|
logs/
|
||||||
|
|
||||||
|
# Операционные системы
|
||||||
|
.DS_Store
|
||||||
|
.DS_Store?
|
||||||
|
._*
|
||||||
|
.Spotlight-V100
|
||||||
|
.Trashes
|
||||||
|
ehthumbs.db
|
||||||
|
Thumbs.db
|
||||||
|
|
||||||
|
# Редакторы
|
||||||
|
.vscode/
|
||||||
|
.idea/
|
||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
*~
|
||||||
|
**.wg0.conf
|
||||||
0
Dockerfile
Normal file
0
Dockerfile
Normal file
125
app.py
Normal file
125
app.py
Normal file
@@ -0,0 +1,125 @@
|
|||||||
|
import os, logging, io, random
|
||||||
|
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup, LabeledPrice
|
||||||
|
from telegram.ext import Application, CommandHandler, CallbackQueryHandler, MessageHandler, PreCheckoutQueryHandler, filters, ContextTypes
|
||||||
|
from PIL import Image, ImageDraw, ImageFont
|
||||||
|
import requests
|
||||||
|
from payments import send_invoice
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
TOKEN = os.getenv('TELEGRAM_TOKEN')
|
||||||
|
OLLAMA_URL = os.getenv('OLLAMA_URL', 'http://localhost:11434')
|
||||||
|
BALANCES = {} # В проде → Redis
|
||||||
|
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
MEME_TEMPLATES = ['drake.jpg', 'this-is-fine.jpg', 'distracted-bf.jpg']
|
||||||
|
|
||||||
|
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
keyboard = [
|
||||||
|
[InlineKeyboardButton("🎨 Создать мем (25⭐)", callback_data='buy_meme')],
|
||||||
|
[InlineKeyboardButton("💰 Баланс", callback_data='balance')]
|
||||||
|
]
|
||||||
|
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||||
|
await update.message.reply_text(
|
||||||
|
'😺 <b>IT Meme Pro Bot</b>\n\n'
|
||||||
|
'Генерирую IT-мемы за <b>25 Telegram Stars</b>!\n'
|
||||||
|
'Коты в проде, Kubernetes фейлы, DevOps драма\n\n'
|
||||||
|
'Нажмите кнопку!',
|
||||||
|
reply_markup=reply_markup, parse_mode='HTML')
|
||||||
|
|
||||||
|
async def generate_meme(prompt: str):
|
||||||
|
# Ollama текст
|
||||||
|
url = f"{OLLAMA_URL}/api/generate"
|
||||||
|
data = {
|
||||||
|
"model": "qwen2.5:3b",
|
||||||
|
"prompt": f"""Создай короткий смешной текст IT-мема (макс 12 слов)
|
||||||
|
на тему: "{prompt}". Стиль: коты в продакшене, devops-проблемы. Только текст!""",
|
||||||
|
"stream": False
|
||||||
|
}
|
||||||
|
resp = requests.post(url, json=data, timeout=30).json()
|
||||||
|
meme_text = resp['response'].strip()
|
||||||
|
|
||||||
|
# Pillow изображение
|
||||||
|
template = Image.open(f'templates/{random.choice(MEME_TEMPLATES)}')
|
||||||
|
draw = ImageDraw.Draw(template)
|
||||||
|
|
||||||
|
try:
|
||||||
|
font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 45)
|
||||||
|
except:
|
||||||
|
font = ImageFont.load_default()
|
||||||
|
|
||||||
|
# Текст с обводкой
|
||||||
|
bbox = draw.textbbox((0, 0), meme_text, font=font)
|
||||||
|
width, height = bbox[2] - bbox[0], bbox[3] - bbox[1]
|
||||||
|
x, y = (template.width - width) // 2, template.height - height - 60
|
||||||
|
|
||||||
|
# Обводка
|
||||||
|
for dx, dy in [(-2,0),(2,0),(0,-2),(0,2)]:
|
||||||
|
draw.text((x+dx, y+dy), meme_text, font=font, fill='black')
|
||||||
|
draw.text((x, y), meme_text, font=font, fill='white')
|
||||||
|
|
||||||
|
bio = io.BytesIO()
|
||||||
|
template.save(bio, 'PNG')
|
||||||
|
bio.seek(0)
|
||||||
|
return bio, meme_text
|
||||||
|
|
||||||
|
async def button_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
query = update.callback_query
|
||||||
|
await query.answer()
|
||||||
|
|
||||||
|
if query.data == 'balance':
|
||||||
|
bal = BALANCES.get(query.from_user.id, 0)
|
||||||
|
await query.edit_message_text(f'💰 Баланс: <b>{bal} мемов</b>', parse_mode='HTML')
|
||||||
|
|
||||||
|
elif query.data == 'buy_meme':
|
||||||
|
await send_invoice(update.callback_query.message, context)
|
||||||
|
|
||||||
|
async def precheckout_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
query = update.pre_checkout_query
|
||||||
|
await query.answer(ok=True)
|
||||||
|
|
||||||
|
async def successful_payment_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
user_id = update.effective_user.id
|
||||||
|
BALANCES[user_id] = BALANCES.get(user_id, 0) + 1 # +1 мем
|
||||||
|
|
||||||
|
await update.message.reply_text(
|
||||||
|
'✅ <b>Оплата прошла! Баланс: 1 мем</b>\n\n'
|
||||||
|
'🎨 Отправьте тему мема:\n"kubernetes pod pending"\n"nfs mount failed"',
|
||||||
|
parse_mode='HTML'
|
||||||
|
)
|
||||||
|
|
||||||
|
async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
user_id = update.effective_user.id
|
||||||
|
balance = BALANCES.get(user_id, 0)
|
||||||
|
|
||||||
|
if balance <= 0:
|
||||||
|
await update.message.reply_text('❌ Нет мемов! Купите: /start')
|
||||||
|
return
|
||||||
|
|
||||||
|
await update.message.reply_text('🎨 Генерирую...')
|
||||||
|
try:
|
||||||
|
meme_img, meme_text = await generate_meme(update.message.text)
|
||||||
|
BALANCES[user_id] -= 1
|
||||||
|
|
||||||
|
await update.message.reply_photo(
|
||||||
|
photo=meme_img,
|
||||||
|
caption=f'😹 <b>Ваш IT-мем готов!</b>\n\n"{meme_text}"\n\n💰 Осталось: {BALANCES[user_id]} мемов',
|
||||||
|
parse_mode='HTML'
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Meme error: {e}")
|
||||||
|
await update.message.reply_text('❌ Ошибка. Попробуйте позже.')
|
||||||
|
|
||||||
|
def main():
|
||||||
|
app = Application.builder().token(TOKEN).build()
|
||||||
|
app.add_handler(CommandHandler("start", start))
|
||||||
|
app.add_handler(CallbackQueryHandler(button_handler))
|
||||||
|
app.add_handler(PreCheckoutQueryHandler(precheckout_callback))
|
||||||
|
app.add_handler(MessageHandler(filters.SUCCESSFUL_PAYMENT, successful_payment_callback))
|
||||||
|
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_text))
|
||||||
|
app.run_polling()
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
||||||
0
k8s/01-namespace.yaml
Normal file
0
k8s/01-namespace.yaml
Normal file
0
k8s/02-ollama.yaml
Normal file
0
k8s/02-ollama.yaml
Normal file
0
k8s/03-bot.yaml
Normal file
0
k8s/03-bot.yaml
Normal file
0
k8s/04-secrets.yaml
Normal file
0
k8s/04-secrets.yaml
Normal file
0
payments.py
Normal file
0
payments.py
Normal file
4
requirements.txt
Normal file
4
requirements.txt
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
python-telegram-bot==20.7
|
||||||
|
pillow==10.4.0
|
||||||
|
requests==2.32.3
|
||||||
|
python-dotenv==1.0.1
|
||||||
Reference in New Issue
Block a user