def test_get_curr_time_diff(self): start = arrow.Arrow(2017, 8, 14, 12, 0) end = arrow.Arrow(2017, 8, 14, 13, 0) self.assertEqual(ArrowUtil.get_curr_time_diff(start=start, stop=end), 60) self.assertEqual( ArrowUtil.get_curr_time_diff(start=start, stop=end, base_hour=True), 1 )
def test_is_between(self): self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 12, 0)), True) self.assertEqual( ArrowUtil.is_between((0, 0), (24, 0), now=arrow.Arrow(2017, 8, 14, 12, 0)), True) self.assertEqual( ArrowUtil.is_between((10, 0), (24, 0), now=arrow.Arrow(2017, 8, 14, 9, 0)), False) self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 10, 0)), True) self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 20, 0)), True)
def total_chart(self): records = [] for i in range(-6, 1, 1): records.append(self.__get_total_score(i)) date = [-6, -5, -4, -3, -2, -1, 0] x_ticks = ArrowUtil.format_weekly_date() legend = self.column_list data = [] for l in legend: data.append(list(map(lambda x: x[l], records))) f_name = "total_weekly_report.png" title = "Total Report" Plot.make_line( date, data, f_name, legend=legend, x_ticks=x_ticks, x_label="Total Point", y_label="Days", title=title, ) self.slackbot.file_upload(f_name, title=title, comment=MsgResource.TOTAL_REPORT)
def total_chart(self): records = [] for i in range(-6, 1, 1): records.append(self.__get_total_score(i)) date = [-6, -5, -4, -3, -2, -1, 0] x_ticks = ArrowUtil.format_weekly_date() legend = self.column_list data = [] for l in legend: data.append(list(map(lambda x: x[l], records))) f_name = "total_weekly_report.png" title = "Total Report" Plot.make_line( date, data, f_name, legend=legend, x_ticks=x_ticks, x_label="Total Point", y_label="Days", title=title, ) self.slackbot.file_upload(f_name, title=title, comment=MsgResource.TOTAL_REPORT)
def is_holiday(self): record = self.data_handler.read_record() holiday = record.get("Holiday", None) if holiday is None: return not ArrowUtil.is_weekday() else: return holiday
def is_holiday(self): record = self.data_handler.read_record() holiday = record.get("Holiday", None) if holiday is None: return not ArrowUtil.is_weekday() else: return holiday
def commit(self, timely="daily"): events = self.github.get_user(self.username).get_events() if isinstance(timely, int): point_start = self.__time_point(timely) point_end = self.__time_point(timely + 1) commit_count = self.__get_event_count(events, point_start, point_end) return commit_count elif timely == "daily": point_start = self.__time_point(0) point_end = self.__time_point(1) commit_count = self.__get_event_count(events, point_start, point_end) if commit_count == 0: self.slackbot.send_message( text=MsgResource.GITHUB_COMMIT_EMPTY) else: self.slackbot.send_message( text=MsgResource.GITHUB_COMMIT_EXIST( commit_count=commit_count)) elif timely == "weekly": commit_count_list = [] for i in range(-6, 1, 1): record = self.data_handler.read_record(days=i) commit_count_list.append(record.get("Github", 0)) date = [-6, -5, -4, -3, -2, -1, 0] x_ticks = ArrowUtil.format_weekly_date() y_ticks = [i for i in range(max(commit_count_list) + 1)] f_name = "github_weekly_commit.png" title = "Github Commit" Plot.make_bar( date, commit_count_list, f_name, x_ticks=x_ticks, y_ticks=y_ticks, x_label="Commit Count", title=title, ) self.slackbot.file_upload(f_name, title=title, comment=MsgResource.GITHUB_COMMIT_WEEKLY) elif timely == "ten_days": commit_count_list = [] for i in range(-9, 1, 1): point_start = self.__time_point(i) point_end = self.__time_point(i + 1) commit_count_list.append( self.__get_event_count(events, point_start, point_end)) return commit_count_list
def commit(self, timely="daily"): events = self.github.get_user(self.username).get_events() if isinstance(timely, int): point_start = self.__time_point(timely) point_end = self.__time_point(timely + 1) commit_count = self.__get_event_count(events, point_start, point_end) return commit_count elif timely == "daily": point_start = self.__time_point(0) point_end = self.__time_point(1) commit_count = self.__get_event_count(events, point_start, point_end) if commit_count == 0: self.slackbot.send_message(text=MsgResource.GITHUB_COMMIT_EMPTY) else: self.slackbot.send_message( text=MsgResource.GITHUB_COMMIT_EXIST(commit_count=commit_count) ) elif timely == "weekly": commit_count_list = [] for i in range(-6, 1, 1): record = self.data_handler.read_record(days=i) commit_count_list.append(record.get("Github", 0)) date = [-6, -5, -4, -3, -2, -1, 0] x_ticks = ArrowUtil.format_weekly_date() y_ticks = [i for i in range(max(commit_count_list) + 1)] f_name = "github_weekly_commit.png" title = "Github Commit" Plot.make_bar( date, commit_count_list, f_name, x_ticks=x_ticks, y_ticks=y_ticks, x_label="Commit Count", title=title, ) self.slackbot.file_upload( f_name, title=title, comment=MsgResource.GITHUB_COMMIT_WEEKLY ) elif timely == "ten_days": commit_count_list = [] for i in range(-9, 1, 1): point_start = self.__time_point(i) point_end = self.__time_point(i + 1) commit_count_list.append( self.__get_event_count(events, point_start, point_end) ) return commit_count_list
def test_is_between(self): self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 12, 0)), True, ) self.assertEqual( ArrowUtil.is_between((0, 0), (24, 0), now=arrow.Arrow(2017, 8, 14, 12, 0)), True, ) self.assertEqual( ArrowUtil.is_between((10, 0), (24, 0), now=arrow.Arrow(2017, 8, 14, 9, 0)), False, ) self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 10, 0)), True, ) self.assertEqual( ArrowUtil.is_between((10, 0), (20, 0), now=arrow.Arrow(2017, 8, 14, 20, 0)), True, )
def test_is_today_day_of_week(self): today_day_of_week = arrow.now().weekday() + 1 self.assertEqual(ArrowUtil.is_today_day_of_week([0]), True) self.assertEqual(ArrowUtil.is_today_day_of_week([today_day_of_week]), True) self.assertEqual( ArrowUtil.is_today_day_of_week([today_day_of_week, 4, 5]), True ) self.assertEqual(ArrowUtil.is_today_day_of_week([10]), False) if ArrowUtil.is_weekday(): self.assertEqual(ArrowUtil.is_today_day_of_week([8]), True) else: self.assertEqual(ArrowUtil.is_today_day_of_week([9]), True)
def edit_attention(self, category, data, days=0): record = self.read_record(days=days) activity_data = record.get("activity", {}) assert category in activity_data latest_data = activity_data[category][-1] task_end_time = arrow.get(latest_data["end_time"]) current_time = data["time"] if ArrowUtil.get_curr_time_diff(task_end_time, current_time) < 40: latest_data["score"] = data["score"] else: pass self.edit_record(("activity", activity_data), days=days)
def edit_attention(self, category, data, days=0): record = self.read_record(days=days) activity_data = record.get("activity", {}) assert category in activity_data latest_data = activity_data[category][-1] task_end_time = arrow.get(latest_data["end_time"]) current_time = data["time"] if ArrowUtil.get_curr_time_diff(task_end_time, current_time) < 40: latest_data["score"] = data["score"] else: pass self.edit_record(("activity", activity_data), days=days)
def total_score(self): today_data = self.__get_total_score() self.data_handler.edit_summary(today_data) color = MsgResource.SCORE_COLOR(today_data["total"]) today_data["Color"] = color yesterday_data = self.__get_total_score(-1) for k, v in today_data.items(): if isinstance(v, float): y_point = yesterday_data.get(k, False) if not y_point: continue else: diff = v - float(y_point) diff = round(diff * 100) / 100 if diff > 0: diff = "+" + str(diff) else: diff = str(diff) today_data[k] = str(v) + " (" + diff + ")" elif isinstance(v, bool): if v: today_data[k] = "O" else: today_data[k] = "X" record = self.data_handler.read_record() activity = record.get("activity", {}) # Sleep Time go_to_bed = activity.get("go_to_bed", None) wake_up = activity.get("wake_up", None) if go_to_bed is not None and wake_up is not None: go_to_bed_time = arrow.get(go_to_bed) wake_up_time = arrow.get(wake_up) sleep_hour = ArrowUtil.get_curr_time_diff(start=go_to_bed_time, stop=wake_up_time, base_hour=True) today_data["Sleep"] = (go_to_bed_time.format("HH:mm") + " ~ " + wake_up_time.format("HH:mm") + " : " + str(sleep_hour) + "h (" + str(today_data["Sleep"]) + ")") # Working Hour in_company = activity.get("in_company", None) out_company = activity.get("out_company", None) if in_company is not None and out_company is not None: in_company_time = arrow.get(in_company) out_company_time = arrow.get(out_company) working_hour = ArrowUtil.get_curr_time_diff(start=in_company_time, stop=out_company_time, base_hour=True) today_data["Working Hour"] = (in_company_time.format("HH:mm") + " ~ " + out_company_time.format("HH:mm") + " : " + str(working_hour) + "h") attachments = MsgTemplate.make_summary_template(today_data) self.slackbot.send_message(attachments=attachments)
def total_score(self): today_data = self.__get_total_score() self.data_handler.edit_summary(today_data) color = MsgResource.SCORE_COLOR(today_data["total"]) today_data["Color"] = color yesterday_data = self.__get_total_score(-1) for k, v in today_data.items(): if isinstance(v, float): y_point = yesterday_data.get(k, False) if not y_point: continue else: diff = v - float(y_point) diff = round(diff * 100) / 100 if diff > 0: diff = "+" + str(diff) else: diff = str(diff) today_data[k] = str(v) + " (" + diff + ")" elif isinstance(v, bool): if v: today_data[k] = "O" else: today_data[k] = "X" record = self.data_handler.read_record() activity = record.get("activity", {}) # Sleep Time go_to_bed = activity.get("go_to_bed", None) wake_up = activity.get("wake_up", None) if go_to_bed is not None and wake_up is not None: go_to_bed_time = arrow.get(go_to_bed) wake_up_time = arrow.get(wake_up) sleep_hour = ArrowUtil.get_curr_time_diff( start=go_to_bed_time, stop=wake_up_time, base_hour=True ) today_data["Sleep"] = ( go_to_bed_time.format("HH:mm") + " ~ " + wake_up_time.format("HH:mm") + " : " + str(sleep_hour) + "h (" + str(today_data["Sleep"]) + ")" ) # Working Hour in_company = activity.get("in_company", None) out_company = activity.get("out_company", None) if in_company is not None and out_company is not None: in_company_time = arrow.get(in_company) out_company_time = arrow.get(out_company) working_hour = ArrowUtil.get_curr_time_diff( start=in_company_time, stop=out_company_time, base_hour=True ) today_data["Working Hour"] = ( in_company_time.format("HH:mm") + " ~ " + out_company_time.format("HH:mm") + " : " + str(working_hour) + "h" ) attachments = MsgTemplate.make_summary_template(today_data) self.slackbot.send_message(attachments=attachments)