4563e383962690cc196f4551f217d488501b660e,bin/count_users_in_rooms.py,,,#,38

Before Change

    dbuser = config["database"]["user"]
    dbpass = config["database"]["password"]

    try:
        conn = psycopg2.connect("dbname="%s" user="%s" host="%s" port="%s" password="%s"" % (
            dbname, dbuser, dbhost, dbport, dbpass)
        )
    except:
        raise RuntimeError("could not connect to db")

    cur = conn.cursor()
    cur.execute(select count(*) from rooms_users_association_table)
    the_count = cur.fetchone()[0]

After Change

    if dbtype == "rdbms" and dbdriver.startswith("mysql"):
        import MySQLdb

        conn = MySQLdb.connect(passwd=dbpass, db=dbname, user=dbuser, host=dbhost, port=dbport)
        cur = conn.cursor()
        cur.execute(select count(*) from rooms_users_association_table)
        the_count = cur.fetchone()[0]

r_host, r_port = config["cache"]["host"].split(":")
r_db = config["cache"]["db"]

Instances