mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-04-13 09:42:01 +00:00
139 lines
4 KiB
Python
139 lines
4 KiB
Python
from flask import Flask, abort
|
|
import os
|
|
import psycopg2
|
|
import pymysql
|
|
import pymongo
|
|
import redis
|
|
import pyes
|
|
|
|
if os.getenv("DOCKERBUILD") == "True":
|
|
import newsblur_web.docker_local_settings as settings
|
|
else:
|
|
import flask_monitor.flask_settings as settings
|
|
|
|
app = Flask(__name__)
|
|
|
|
@app.route("/db_check/postgres")
|
|
def db_check_postgres():
|
|
connect_params = "dbname='%s' user='%s' password='%s' host='%s' port='%s'" % (
|
|
settings.DATABASES['default']['NAME'],
|
|
settings.DATABASES['default']['USER'],
|
|
settings.DATABASES['default']['PASSWORD'],
|
|
settings.DATABASES['default']['HOST'],
|
|
settings.DATABASES['default']['PORT'],
|
|
)
|
|
try:
|
|
conn = psycopg2.connect(connect_params)
|
|
except:
|
|
print(" ---> Postgres can't connect to the database: %s" % connect_params)
|
|
abort(502)
|
|
|
|
cur = conn.cursor()
|
|
cur.execute("""SELECT id FROM feeds ORDER BY feeds.id DESC LIMIT 1""")
|
|
rows = cur.fetchall()
|
|
for row in rows:
|
|
return str(row[0])
|
|
|
|
abort(404)
|
|
|
|
@app.route("/db_check/mysql")
|
|
def db_check_mysql():
|
|
connect_params = "dbname='%s' user='%s' password='%s' host='%s' port='%s'" % (
|
|
settings.DATABASES['default']['NAME'],
|
|
settings.DATABASES['default']['USER'],
|
|
settings.DATABASES['default']['PASSWORD'],
|
|
settings.DATABASES['default']['HOST'],
|
|
settings.DATABASES['default']['PORT'],
|
|
)
|
|
try:
|
|
|
|
conn = pymysql.connect(host=settings.DATABASES['default']['HOST'],
|
|
port=settings.DATABASES['default']['PORT'],
|
|
user=settings.DATABASES['default']['USER'],
|
|
passwd=settings.DATABASES['default']['PASSWORD'],
|
|
db=settings.DATABASES['default']['NAME'])
|
|
except:
|
|
print(" ---> Mysql can't connect to the database: %s" % connect_params)
|
|
abort(502)
|
|
|
|
cur = conn.cursor()
|
|
cur.execute("""SELECT id FROM feeds ORDER BY feeds.id DESC LIMIT 1""")
|
|
rows = cur.fetchall()
|
|
for row in rows:
|
|
return str(row[0])
|
|
|
|
abort(404)
|
|
|
|
@app.route("/db_check/mongo")
|
|
def db_check_mongo():
|
|
try:
|
|
client = pymongo.MongoClient('mongodb://%s' % settings.MONGO_DB['host'])
|
|
db = client.newsblur
|
|
except:
|
|
abort(502)
|
|
|
|
return str(db.stories.count())
|
|
|
|
@app.route("/db_check/redis")
|
|
def db_check_redis():
|
|
redis_host = getattr(settings, 'REDIS', {'host': 'db_redis'})
|
|
try:
|
|
r = redis.Redis(redis_host['host'], db=0)
|
|
except:
|
|
abort(502)
|
|
|
|
randkey = r.randomkey()
|
|
if randkey:
|
|
return str(randkey)
|
|
else:
|
|
abort(404)
|
|
|
|
@app.route("/db_check/redis_story")
|
|
def db_check_redis_story():
|
|
redis_host = getattr(settings, 'REDIS', {'host': 'db_redis_story'})
|
|
try:
|
|
r = redis.Redis(redis_host['host'], db=1)
|
|
except:
|
|
abort(502)
|
|
|
|
randkey = r.randomkey()
|
|
if randkey:
|
|
return str(randkey)
|
|
else:
|
|
abort(404)
|
|
|
|
@app.route("/db_check/redis_sessions")
|
|
def db_check_redis_sessions():
|
|
redis_host = getattr(settings, 'REDIS', {'host': 'db_redis_sessions'})
|
|
try:
|
|
r = redis.Redis(redis_host['host'], db=5)
|
|
except:
|
|
abort(502)
|
|
|
|
randkey = r.randomkey()
|
|
if randkey:
|
|
return str(randkey)
|
|
else:
|
|
abort(404)
|
|
|
|
@app.route("/db_check/elasticsearch")
|
|
def db_check_elasticsearch():
|
|
es_host = getattr(settings, 'ELASTICSEARCH_FEED_HOSTS', ['db_search_feed:9200'])
|
|
try:
|
|
conn = pyes.ES(es_host)
|
|
except:
|
|
abort(502)
|
|
|
|
if conn.indices.exists_index('feeds-index'):
|
|
return str("Index exists, but didn't try search")
|
|
# query = pyes.query.TermQuery("title", "daring fireball")
|
|
# results = conn.search(query=query, size=1, doc_types=['feeds-type'], sort="num_subscribers:desc")
|
|
# for result in results:
|
|
# return unicode(result)
|
|
# else:
|
|
# abort(404)
|
|
else:
|
|
abort(404)
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0")
|