def get(self):
username = self.get_secure_cookie('username')
users = None
with db.conn.cursor() as c:
user_id = self.get_secure_cookie('user_id')
r = db.query_one(c, 'SELECT admin FROM users WHERE id = ?', user_id)
admin = bool(r.admin)
if admin:
users = db.query(c, 'SELECT username, admin FROM users')
users = list(users)
self.render('account.html', username=username, admin=admin, users=users)
def get(self):
username = self.get_secure_cookie('username')
users = None
with db.conn.cursor() as c:
user_id = self.get_secure_cookie('user_id')
r = db.query_one(c, 'SELECT admin FROM users WHERE id = ?', user_id)
admin = bool(r.admin)
if admin:
users = db.query(c, 'SELECT username, admin FROM users')
users = list(users)
self.render('account.html', username=username, admin=admin, users=users)
def post(self):
with db.conn.cursor() as c:
r = db.query_one(c, 'SELECT admin FROM users WHERE id = ?', self.current_user)
admin = bool(r.admin)
if not admin:
raise tornado.web.HTTPError(403)
username = self.get_argument('username')
password = self.get_argument('password')
if not username or not password:
raise tornado.web.HTTPError(400)
db.create_user(self.get_secure_cookie('username'), username, password)
self.redirect('/account')
def update_user_deck():
try:
json = request.args["json"]
deck_id = request.args["deck_id"]
cursor = get_cursor()
cursor.execute(
"""UPDATE UserDecks SET json=%s WHERE deck_id=%s""", (json, deck_id))
user_deck_id = cursor.lastrowid
cursor.connection.commit()
cursor.close()
return to_json(query_one("SELECT * FROM UserDecks WHERE deck_id = %s", user_deck_id))
except:
return '{"error":"not saved"}'
def save_user_deck():
try:
json = request.args["json"]
google_id = request.args["google_id"]
cursor = get_cursor()
cursor.execute(
"INSERT INTO UserDecks (google_id,json) VALUES (%s, %s)", (google_id, json))
user_deck_id = cursor.lastrowid
cursor.connection.commit()
cursor.close()
return to_json(query_one("SELECT * FROM UserDecks WHERE deck_id = %s", user_deck_id))
except:
return '{"error":"not saved"}'
def get_or_create_keyword(keyword):
created = False
output = query_one("""
SELECT keyword_id, last_updated FROM Keywords where keyword = %s
""", (keyword,))
if output:
keyword_id, last_updated = output
else:
keyword_id = None
if not keyword_id:
created = True
cursor = get_cursor()
cursor.execute("""
INSERT INTO Keywords (keyword) VALUES (%s)
""", (keyword,))
keyword_id = cursor.lastrowid
last_updated = datetime.now()
cursor.connection.commit()
cursor.close()
return keyword_id, last_updated, created
def zf_download(fid):
if fid is None:
return 'file not found'
row = db.query_one('select imgs from house where id=?', fid)
if row is None:
return 'file not found'
file_list = row['imgs'].split(',')
dl_name = '{}.zip'.format(fid)
memory_file = BytesIO()
with zipfile.ZipFile(memory_file, "w", zipfile.ZIP_DEFLATED) as zf:
i = 1
for _file in file_list:
with open(os.path.join(os.getcwd(), 'data/images/', _file),
'rb') as fp:
zf.writestr('{}.jpg'.format(i), fp.read())
i = i + 1
memory_file.seek(0)
return send_file(memory_file,
attachment_filename=dl_name,
as_attachment=True)
def ping():
return text(db.query_one('select version()')['version'])
def ping():
return text(db.query_one("select version()")["version"])
def keyword_has_decks(keyword_id):
res = query_one(
"SELECT COUNT(*) FROM KeywordsQuizletDecks where keyword_id = %s", (keyword_id,))
return res[0] > 1 if res else False
def helo(self):
with db.conn.cursor() as c:
r = db.query_one(c, 'SELECT json from maps')
self.write_message('MAP ' + r.json)
def helo(self):
with db.conn.cursor() as c:
r = db.query_one(c, 'SELECT json from maps')
self.write_message('MAP ' + r.json)
def publish_house(fid, tid):
num = db.query_count(
'select count(*) from publish_record where fid=? and status!=?', fid,
'待发布')
if num > 0:
print('house is already published.', fid)
db.close()
return
f_row = db.query_one('select * from house where id=?', fid)
t_row = db.query_one('select * from template where id=?', tid)
if f_row is None:
print('house not found.', fid)
db.close()
return
if t_row is None:
print('template not found.', tid)
db.close()
return
db.execute('update publish_record set status=? where fid=?', '发布中', fid)
options = Options()
driver_path = './drivers/chromedriver_mac'
if sys.platform.startswith('linux'):
options.add_argument('--no-sandbox')
options.add_argument('window-size=1920x3000')
options.add_argument('--disable-gpu')
options.add_argument('--hide-scrollbars')
options.add_argument('blink-settings=imagesEnabled=false')
options.add_argument('--headless')
driver_path = './drivers/chromedriver_linux'
broswer = webdriver.Chrome(os.path.abspath(driver_path), options=options)
broswer.set_page_load_timeout(15)
broswer.set_script_timeout(20)
try:
print('try opening website...')
broswer.get('https://vip.anjuke.com/login/')
except Exception as ex:
print('timeout...stop page loading: ', ex.__str__())
broswer.execute_script("window.stop()")
try:
# 切换账号密码登录
span = broswer.find_element_by_xpath(
'/html/body/div[4]/div[2]/div/div[1]/span[2]')
span.click()
time.sleep(1)
# 登录
print('try login...')
broswer.find_element_by_id('loginName').send_keys('13928753569')
broswer.find_element_by_id('loginPwd').send_keys('YZXyzx123456')
broswer.find_element_by_id('loginSubmit').submit()
time.sleep(1)
# 发布房源
print('switch to publish rent page')
broswer.get('https://vip.anjuke.com/house/publish/rent/?from=manage')
time.sleep(2)
# 阅读条款
print('readed button')
readBtn = broswer.find_element_by_xpath(
'//*[@id="publishHouse"]/div[2]/div/button')
if readBtn is not None:
readBtn.click()
# 小区名称
print('fill community unite...')
broswer.find_element_by_id('community_unite').send_keys(
t_row['community_unite'])
time.sleep(3)
broswer.find_element_by_xpath(
'//*[@id="publish_form"]/div[3]/div/ul/li[1]').click()
# 室
print('fill shi...')
broswer.find_element_by_name('shi').send_keys(t_row['shi'])
# 厅
print('fill ting...')
broswer.find_element_by_name('ting').send_keys(t_row['ting'])
# 卫
print('fill wei...')
broswer.find_element_by_name('wei').send_keys(t_row['wei'])
# 所在楼层
print('fill current floor')
broswer.find_element_by_name('suoZaiLouCeng').send_keys(
t_row['suoZaiLouCeng'])
# 总共楼层
print('fill total floors')
broswer.find_element_by_name('zongLouCeng').send_keys(
t_row['zongLouCeng'])
# 电梯
print('select dianti')
broswer.find_element_by_xpath(
'//*[@id="publish_form"]/div[8]/label[2]').click()
# 总面积
broswer.find_element_by_name('mianJi').send_keys(t_row['mianJi'])
# 已出租
broswer.find_element_by_name('params_101').send_keys(
t_row['currentrent'])
# 共几户
broswer.find_element_by_name('params_196').send_keys(
t_row['totalrent'])
# 房屋类型
print('selectting house type...')
broswer.find_element_by_xpath(
'//*[@id="select-housetype"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-housetype"]/div/ul/li[3]').click()
# 装修情况
print('selectting house fit...')
broswer.find_element_by_xpath('//*[@id="select-housefit"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-housefit"]/div/ul/li[4]').click()
# 朝向
print('selectting exposure')
broswer.find_element_by_xpath('//*[@id="select-exposure"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-exposure"]/div/ul/li[3]').click()
# 公共设施
broswer.find_element_by_xpath(
'//*[@id="publish_form"]/div[12]/div/div/label[1]').click()
# 待租卧室 - 类型
print('select bedroom')
broswer.find_element_by_xpath('//*[@id="select-bedroom"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-bedroom"]/div/ul/li[3]').click()
# 待租卧室 - 朝向
broswer.find_element_by_xpath(
'//*[@id="select-roomorient"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-roomorient"]/div/ul/li[3]').click()
# 待租卧室 - 限制
broswer.find_element_by_xpath('//*[@id="select-sex"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-sex"]/div/ul/li[2]').click()
# 待租卧室 - 面积
broswer.find_element_by_name('params_12').send_keys('23')
# 价格
print('fill price...')
broswer.find_element_by_name('jiaGe').send_keys(t_row['jiaGe'])
# 付款方式
print('fill pay type...')
broswer.find_element_by_xpath('//*[@id="select-paymode"]/div').click()
time.sleep(0.2)
broswer.find_element_by_xpath(
'//*[@id="select-paymode"]/div/ul/li[2]').click()
# 标题
print('fill title...')
broswer.find_element_by_name('title').send_keys(t_row['title'])
# 详细介绍
# broswer.find_element_by_name('content_fangyuanxiangqing').send_keys('这里是描述')
# broswer.execute_script('document.getElementsByName("content_fangyuanxiangqing").value = "这里是描述"')
# 选择房源模板
print('openning house templete window...')
broswer.find_element_by_xpath('//*[@id="template_ctrl"]/a[2]').click()
time.sleep(2)
print('select house templete...')
broswer.find_element_by_xpath(
'//*[@id="house_template"]/div[2]/div[1]/dl/dd/label').click()
print('close templete window')
broswer.find_element_by_xpath(
'//*[@id="house_template"]/div[2]/div[2]/a[1]').click()
print('uploading images...')
# 上传室内图
for img in str(f_row['imgs']).split(','):
file_path = os.path.abspath(os.path.join('./data/images/', img))
broswer.find_element_by_id('room_fileupload').send_keys(file_path)
# 上传户型图
file_path = os.path.abspath('./data/house_model.jpg')
broswer.find_element_by_id('model-fileupload').send_keys(file_path)
# 等待图片上传完成
print('waiting for upload images')
time.sleep(20)
# 设置封面
broswer.find_element_by_xpath(
'//*[@id="room-upload-display"]/div[1]/div/i[1]').click()
time.sleep(2)
# 发布房源
print('publish rent info...')
broswer.find_element_by_id('publish-rent-add').click()
time.sleep(5)
# 结果检测
result = broswer.find_element_by_xpath(
'/html/body/div[4]/div[2]/dl/dd')
if result is not None and result.text == '操作成功':
print('发布成功')
db.execute('update publish_record set status=? where fid=?', '已发布',
fid)
else:
print('发布失败')
db.execute('update publish_record set status=? where fid=?',
'发布失败', fid)
db.close()
time.sleep(3)
broswer.quit()
except Exception as ex:
print(ex.__str__())
broswer.quit()
