コード例 #1
0
ファイル: diary.py プロジェクト: hizzgdev/walpw
 def __init__(self):
     self.dda = DiaryDao()
     self.diary_bucket = Bucket("diary")
コード例 #2
0
ファイル: diary.py プロジェクト: hizzgdev/walpw
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