def __init__(self):
self.dda = DiaryDao()
self.diary_bucket = Bucket("diary")
class DiaryService(object):
def __init__(self):
self.dda = DiaryDao()
self.diary_bucket = Bucket("diary")
def dump_users(self):
users = self.dda.get_users()
if users != None:
for u in users:
# obj_id = '/user/{0}.dat'.format(u['girl'])
# obj_data = {'password': u['password'], 'diaryId': u['diaryId'], 'boy': u['boy'], 'girl': u['girl']}
# self.diary_bucket.put_object(obj_id, json.dumps(obj_data), content_type='application/json', content_encoding='utf-8')
diarys = self.dda.get_user_diarys(u["diaryId"])
obj_d_id = "/diary/{0}.dat".format(u["diaryId"])
self.diary_bucket.put_object(
obj_d_id, json.dumps(diarys), content_type="application/json", content_encoding="utf-8"
)
else:
pass
def get_recent_diarys(self, username, limit):
obj_id = "/user/{0}.dat".format(username)
user = self.diary_bucket.get_object_contents(obj_id)
obj_id = "/diary/{0}.dat".format(u["diaryId"])
return self.dda.get_diarys_by_user_since_date(username, since, limit)
def get_diary_by_id_xml(self, username, diaryid):
if username == None:
return None
diarys = self.dda.get_diary_by_id(diaryid)
if len(diarys) > 0:
diary = diarys[0]
if diary["FDUName"] == username or diary["FDULoverName"] == username or diary["FDOpenLevel"] == 0:
self.dda.hit_diary(diaryid)
else:
diarys = None
return self.strf_diarys("data", diarys)
def get_diary_by_id_dict(self, username, diaryid):
if username == None:
return None
diarys = self.dda.get_diary_by_id(diaryid)
diary = None
if len(diarys) > 0:
diary = diarys[0]
if diary["FDUName"] == username or diary["FDULoverName"] == username or diary["FDOpenLevel"] == 0:
self.dda.hit_diary(diaryid)
else:
diary = None
return diary
def get_diarys_by_date(self, username, year, month, day):
if username == None:
return None
date = datetime.date(year, month, day)
diarys = self.dda.get_diarys_by_user_date(username, date)
for diary in diarys:
self.dda.hit_diary(diary["FDID"])
diarys_str = self.strf_diarys("data", diarys)
return diarys_str
def _get_diarys_by_month(self, username, year, month):
date_start = datetime.date(year, month, 1)
date_end = date_start + relativedelta(months=1)
diarys = self.dda.get_diarysdate_by_user_datescope(username, date_start, date_end)
return diarys
def get_diarysdate_by_month(self, username, year, month):
strArray = []
diarys = self._get_diarys_by_month(username, year, month)
if diarys != None:
for diary in diarys:
strArray.append(diary["FDTime"].strftime("%d"))
return ",".join(strArray)
def get_diarysdate_by_scope(self, year, month):
dateformat = "{:0>4d}"
if year == None:
date_group_count = self.dda.year_group_count()
elif month == None:
date_s = datetime.datetime(year, 1, 1, 0, 0, 0)
date_e = date_s + relativedelta(years=1)
dateformat = "{:0>2d}"
date_group_count = self.dda.month_group_count(date_s, date_e)
else:
date_s = datetime.datetime(year, month, 1, 0, 0, 0)
date_e = date_s + relativedelta(months=1)
dateformat = "{:0>2d}"
date_group_count = self.dda.date_group_count(date_s, date_e)
strdate = []
strcount = []
if date_group_count == None or len(date_group_count) == 0:
return "|"
for row in date_group_count:
strdate.append(dateformat.format(row["d"]))
strcount.append(str(row["c"]))
return ",".join(strdate) + "|" + ",".join(strcount)
def get_diarysdate_by_user_scope(self, username, year, month):
if username == None or len(username) == 0:
return "|"
dateformat = "{:0>4d}"
if year == None:
date_group_count = self.dda.year_group_count(username=username)
elif month == None:
date_s = datetime.datetime(year, 1, 1, 0, 0, 0)
date_e = date_s + relativedelta(years=1)
dateformat = "{:0>2d}"
date_group_count = self.dda.month_group_count(date_s, date_e, username=username)
else:
date_s = datetime.datetime(year, month, 1, 0, 0, 0)
date_e = date_s + relativedelta(months=1)
dateformat = "{:0>2d}"
date_group_count = self.dda.date_group_count(date_s, date_e, username=username)
strdate = []
strcount = []
if date_group_count == None or len(date_group_count) == 0:
return "|"
for row in date_group_count:
strdate.append(dateformat.format(row["d"]))
strcount.append(str(row["c"]))
return ",".join(strdate) + "|" + ",".join(strcount)
def get_diarys_by_user_scope(self, username, year, month):
if year == None:
date_s = None
date_e = None
elif month == None:
date_s = datetime.datetime(year, 1, 1, 0, 0, 0)
date_e = date_s + relativedelta(years=1)
else:
date_s = datetime.datetime(year, month, 1, 0, 0, 0)
date_e = date_s + relativedelta(months=1)
diarys = self.dda.get_diarys_by_user_datescope(date_s, date_e, username)
return self.strf_diarys("info", diarys)
def strf_diarys(self, outputtype, diarys):
strArray = []
if diarys != None:
if outputtype == "data":
for diary in diarys:
strArray.append(self.strf_diary_data(diary))
else:
for diary in diarys:
strArray.append(self.strf_diary_info(diary))
formatter = """<?xml version=\"1.0\" encoding=\"utf-8\"?>
<diarys>
{0}
</diarys>"""
return formatter.format("".join(strArray))
def strf_diary_info(self, diary):
if diary == None:
return None
formatter = """<diary>
<id>{0[FDID]}</id>
<title><![CDATA[{0[FDTitle]}]]></title>
<date>{0[FDTime]:%Y-%m-%d}</date>
<author>{0[FDUName]}</author>
<visit>{0[FDVisitCount]}</visit>
<comment>{0[FDCommCount]}</comment>
</diary>"""
return formatter.format(diary)
def strf_diary_data(self, diary):
if diary == None:
return None
formatter = """<diary>
<id>{0[FDID]}</id>
<title><![CDATA[{0[FDTitle]}]]></title>
<date>{0[FDTime]:%Y-%m-%d}</date>
<weather>{0[FDWeather]}</weather>
<author>{0[FDUName]}</author>
<visit>{0[FDVisitCount]}</visit>
<comment>{0[FDCommCount]}</comment>
<content><![CDATA[{0[FDContent]}]]></content>
<openlevel>{0[FDOpenLevel]}</openlevel>
<allowcomment>{0[FDAllowComment]}</allowcomment>
</diary>"""
return formatter.format(diary)
def is_diary_owner(self, diaryid, username):
owners = self.dda.get_diary_owners(diaryid)
if owners == None:
return False
if username == owners[0] or username == owners[1]:
return True
else:
return False
def can_comment(self, diaryid):
diary_info = self.dda.get_diary_fields(["FDAllowComment"], diaryid)
if diary_info == None or len(diary_info) == 0:
return False
if diary_info[0]["FDAllowComment"]:
return True
else:
return False
