main.py 文件源码

def users_total(self):
        '''Query number of users'''
        try:
            con = psycopg2.connect(self.config['RDS_URI'])
            cur = con.cursor()
            cur.execute("SELECT COUNT(*) FROM users")
            con.commit()
            result = cur.fetchone()
            print("The number of users: %s" %result[0])
        except (Exception, psycopg2.DatabaseError) as error:
            print(error)

        finally:
            if con is not None:
                con.close()