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()
