def login(env, params):
result = {}
try:
conn = db.DB()
email = params['email'][0]
pwd = params['pwd'][0]
sql = "select userid,nickname,pwd from user where email=%s"
data = conn.read(sql, (email))
temp = {}
if pwd == data[0]['pwd']:
temp['login'] = '******'
temp['session_id'] = data[0]['userid']
temp['nickname'] = data[0]['nickname']
else:
temp['login'] = '******'
temp['session_id'] = ""
temp['nickname'] = ""
result['data'] = temp
result['ret'] = 0
result['msg'] = 'ok'
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def get_list(env, params):
result = {}
temp = {}
try:
conn = db.DB()
userid = params['session_id'][0]
start_time = params['start_time'][0]
end_time = params['end_time'][0]
start_num = int(params['start_num'][0])
num = int(params['num'][0])
sql = "select c.* from content c where c.id in (select kwc.content_id from key_word_content kwc where kwc.key_word_id in(select ukw.key_word_id from user_key_word ukw where ukw.user_id = (select u.id from user u where u.userid = %s))) and c.update_time < %s limit %s,%s"
data = conn.read(sql, (userid, start_time, start_num, num))
for i in range(len(data)):
data[i]['create_time'] = str(data[i]['create_time'])
data[i]['update_time'] = str(data[i]['update_time'])
result['data'] = data
result['ret'] = 0
result['msg'] = 'ok'
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def key_word_delete(env, params):
result = {}
temp = {}
try:
conn = db.DB()
userid = params['session_id'][0]
key_word_id = params['id'][0]
sql0 = "select id from user where user.userid=%s"
data0 = conn.read(sql0, (userid))
user_id = data0[0]['id']
sql1 = "delete from user_key_word where user_id=%s and key_word_id=%s"
data1 = conn.write(sql1, (user_id, key_word_id))
if data1 == 1:
result['msg'] = 'ok'
result['ret'] = 0
else:
result['msg'] = 'fail'
result['ret'] = 1
result['data'] = temp
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def key_word_update(env, params):
result = {}
temp = {}
try:
conn = db.DB()
userid = params['session_id'][0]
key_word_id = params['id'][0]
kw = params['kw'][0]
kw_type = params['type'][0]
update_time = time.strftime('%Y-%m-%d %H:%M:%S')
sql = "update key_word set kw=%s,type=%s,update_time=%s where id=%s"
data = conn.write(sql, (kw, kw_type, update_time, key_word_id))
if data == 1:
result['msg'] = 'ok'
result['ret'] = 0
else:
result['msg'] = 'fail'
result['ret'] = 1
result['data'] = temp
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def key_word_add(env, params):
result = {}
temp = {}
try:
conn = db.DB()
userid = params['session_id'][0]
kw = params['kw'][0]
kw_type = params['type'][0]
create_time = time.strftime('%Y-%m-%d %H:%M:%S')
update_time = create_time
sql0 = "select id from user where user.userid=%s"
data0 = conn.read(sql0, (userid))
user_id = data0[0]['id']
sql = "select * from key_word where key_word.kw=%s"
data = conn.read(sql, (kw))
if (len(data) == 0):
print 'data = 0'
sql1 = "insert into key_word(kw,type,create_time,update_time) values(%s,%s,%s,%s)"
data1 = conn.write(sql1, (kw, kw_type, create_time, update_time))
if data1 == 1:
sql2 = "select max(id) from key_word"
data2 = conn.read(sql2)
key_word_id = data2[0]['max(id)']
print key_word_id
else:
key_word_id = data[0]['id']
sql4 = "select * from user_key_word where user_key_word.user_id=%s and user_key_word.key_word_id=%s"
data4 = conn.read(sql4, (user_id, key_word_id))
if (len(data4) == 0):
sql3 = "insert into user_key_word(user_id,key_word_id) values(%s,%s)"
data3 = conn.write(sql3, (user_id, key_word_id))
if data3 == 1:
result['msg'] = 'ok'
result['ret'] = 0
temp['id'] = key_word_id
else:
result['msg'] = 'fail'
result['ret'] = 1
temp['id'] = ''
else:
result['msg'] = 'key word is existed'
result['ret'] = 0
temp['id'] = key_word_id
result['data'] = temp
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def get_all_key_word():
data = None
try:
conn = db.DB()
sql = "select * from key_word"
data = conn.read(sql)
return data
except Exception, e:
print e
return data
def add_content(kw_id, url, title, summary, update_time):
print 'add new content'
try:
conn = db.DB()
create_time = time.strftime('%Y-%m-%d %H:%M:%S')
sql = "insert into content(url,title,summary,create_time,update_time) values(%s,%s,%s,%s,%s)"
data = conn.write(sql, (url, title, summary, create_time, update_time))
if data == 1:
sql1 = "select max(id) from content"
data1 = conn.read(sql1)
content_id = data1[0]['max(id)']
sql2 = "insert into key_word_content(key_word_id,content_id) values(%s,%s)"
data2 = conn.write(sql2, (kw_id, content_id))
if data2 == 1:
print 'success'
else:
print 'error'
else:
print 'error'
except Exception, e:
print e
def key_word_list(env, params):
result = {}
try:
conn = db.DB()
userid = params['session_id'][0]
sql = "select key_word.* from key_word,user,user_key_word where user.id=user_key_word.user_id and key_word.id=user_key_word.key_word_id and user.userid=%s"
data = conn.read(sql, (userid))
for i in range(len(data)):
data[i]['create_time'] = str(data[i]['create_time'])
data[i]['update_time'] = str(data[i]['update_time'])
result['data'] = data
result['ret'] = 0
result['msg'] = 'ok'
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
def register(env, params):
result = {}
temp = {}
try:
conn = db.DB()
userid = str(uuid.uuid1())
nickname = params['nickname'][0]
email = params['email'][0]
pwd = params['pwd'][0]
create_time = time.strftime('%Y-%m-%d %H:%M:%S')
update_time = create_time
sql0 = "select * from user where email=%s"
data0 = conn.read(sql0, (email))
if (len(data0) == 0):
sql = "insert into user(userid,nickname,email,pwd,create_time,update_time) values(%s,%s,%s,%s,%s,%s)"
data = conn.write(
sql, (userid, nickname, email, pwd, create_time, update_time))
if data == 1:
result['msg'] = 'success'
result['ret'] = 0
else:
result['ret'] = 1
result['msg'] = 'sql error'
else:
result['msg'] = 'email is existed'
result['ret'] = 1
result['data'] = temp
json_result = json.dumps(result)
conn.close()
return json_result
except Exception, e:
print e
result['ret'] = 1
result['msg'] = str(e)
json_result = json.dumps(result)
return json_result
data_path = "data/"
cache_path = data_path + "cache/"
# _root = 'www.netbian.com'
_root = 'desk.zol.com.cn/'
_base_url = 'http://' + _root
__filename = util.get_hash(_root)
cache_page_path = cache_path + "pages/" + __filename + "/"
cache_link_path = cache_path + "links/" + __filename + "/"
image_path = data_path + "images/" + __filename + "/"
task_path = data_path + "task/" + __filename + "/"
util.check_dirs((cache_link_path, cache_page_path, task_path))
DB = db.DB()
dstUrl = 'http://' + _root + '/'
next_set = set()
global_list = []
history_list = []
start_time = time.time()
next_list = list(next_set)
next_task = set()
def task_exist():
return file.exist(task_path + "task-exec-" + util.get_hash(__filename) + ".tsk")
def read_task():
