from pymongo import MongoClient


class MongoDB(object):
    conn_addr1 = 'dds-uf6da0fedc9881d41.mongodb.rds.aliyuncs.com:3717'
    conn_addr2 = 'dds-uf6da0fedc9881d42.mongodb.rds.aliyuncs.com:3717'
    replicat_set = 'mgset-12835903'
    username = 'brand'
    password = 'brand'

    def __init__(self, col_name, db_name=None):

        self.client = MongoClient([self.conn_addr1, self.conn_addr2], replicaSet=self.replicat_set)
        self.client.diaoyanbao.authenticate(self.username, self.password)
        if db_name:
            self.mydb = self.client[db_name]
        else:
            self.mydb = self.client['diaoyanbao']
        self.mycol = self.mydb[col_name]

    def find(self, query):
        data = self.mycol.find(query)
        return [x for x in data]


if __name__ == '__main__':
    mongodb = MongoDB('answers')
    query = {'testcaseId': 84}
    datas = mongodb.find(query)
    print(datas[0])