class Match: def __init__(self): self.com = Common() def match_test(self): related = [] file_list = self.com.GetFileList(config.json_store) for i in range(1,6): rand = int(random.random()*len(file_list)) related.append(file_list[rand]) return related
class Search: def __init__(self): self.com = Common() self.json = config.json_store def search_hardmatch(self, name): ''' #usage: A search function for test #arg: search query #return: path of json files which contain search query ''' src = self.json furniture_list = self.com.GetFileList(src) for furniture in furniture_list: if name not in furniture: furniture_list.remove(furniture) return furniture_list
class Insert_db: def __init__(self): self.com = Common() def loadData(self): ''' #usage: Load data form /home/boyang/Documents/furnitures #arg: None #return: Data stored at path above in json structure ''' src = '/home/boyang/Documents/furnitures'; datas = [] furniture_list = self.com.GetFileList(src) for furniture in furniture_list: data = self.com.readJSON(furniture) datas.append(data) return datas def insertData(self, datas): ''' #usage: Insert data into data base #arg: Data you want to insert #rerutn: None ''' db = pymysql.connect(host = '127.0.0.1', user = '******', password = '******', db = 'FurnitoData', charset = 'utf8', cursorclass = pymysql.cursors.DictCursor) cursor = db.cursor() for data in datas: comments = data['reviews'] for comment in comments: sql = 'INSERT INTO comments (furniture_name, comment, vote_up, vote_down) VALUES (%s, %s, %s, %s)' print sql cursor.execute(sql, (data['name'].encode('utf-8'),comment.encode('utf-8'),'0','0')) db.commit() db.close()