pipisa-exporter/exporto.py

60 lines
1.6 KiB
Python
Raw Normal View History

2023-06-20 21:10:04 +10:00
from sqlalchemy import create_engine, Integer, String, BIGINT, Column, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import select, update
2023-06-22 17:37:13 +10:00
from sql_db_data import *
2023-06-20 21:10:04 +10:00
2023-06-20 21:10:25 +10:00
from flask import Flask
app = Flask(__name__)
2023-06-20 21:10:04 +10:00
def get_top_dicks():
session = Session()
2023-06-22 17:37:13 +10:00
top = session.query(User.user_fullname, UserDick.dick_size, User.user_id, UserDick.chat_id).join(UserDick).order_by(UserDick.dick_size.desc()).all()
2023-06-20 21:10:04 +10:00
session.close()
return top
2023-06-21 12:33:09 +10:00
2023-06-21 12:24:29 +10:00
2023-06-21 12:33:09 +10:00
@app.route('/metrics')
def metrics():
metrics = ""
2023-06-21 12:30:26 +10:00
2023-06-21 12:33:09 +10:00
docks = get_top_dicks()
2023-06-21 12:30:26 +10:00
2023-06-21 12:33:09 +10:00
# текущий размер пипис
for dick in docks:
metrics += 'pipisio{user="%s", user_id="%s", chat_id="%s"} %s\n' % (dick[0], dick[2], dick[3], dick[1])
2023-06-21 12:24:29 +10:00
2023-06-21 12:33:09 +10:00
# расстояние до следующего впереди
for k in range(len(docks)):
2023-06-21 12:30:26 +10:00
2023-06-21 12:33:09 +10:00
### не самый красивый код для не самой красивой ситуации
2023-06-21 12:30:26 +10:00
2023-06-21 12:33:09 +10:00
if k == 0:
metrics += 'pipisio_topdick_distanso{user="%s", user_id="%s", chat_id="%s"} %s\n' % (docks[k][0], docks[k][2], docks[k][3], 0)
else:
i = k - 1
upper_dick_size = docks[i][1]
while i > 0 and upper_dick_size == docks[k][1]:
i -= 1
upper_dick_size = docks[i][1]
metrics += 'pipisio_topdick_distanso{user="%s", user_id="%s", chat_id="%s"} %s\n' % (docks[k][0], docks[k][2], docks[k][3], (upper_dick_size - docks[k][1]))
2023-06-21 12:30:26 +10:00
2023-06-21 12:33:09 +10:00
return metrics
2023-06-20 21:10:04 +10:00
2023-06-21 12:33:09 +10:00
if __name__ == '__main__':
app.run(host='0.0.0.0', port=9992)
engine.dispose()