Esempio n. 1
0
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()

        self.username = Config.open_api.github.USERNAME
        self.github = Github(Config.open_api.github.ACCESS_TOKEN)

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot
Esempio n. 2
0
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()

        self.username = Config.open_api.github.USERNAME
        password = Config.open_api.github.PASSWORD
        self.github = Github(self.username, password)

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot
Esempio n. 3
0
class KinoBot:
    def __init__(self) -> None:
        self.slackbot = SlackerAdapter()
        self.logger = Logger().get_logger()
        self.worker = Worker(slackbot=self.slackbot)

        self.error_delay = 1  # Unit (Second)

        # Send a message to channel (init)
        MASTER_NAME = Config.bot.MASTER_NAME
        BOT_NAME = Config.bot.BOT_NAME
        self.slackbot.send_message(
            text=MsgResource.HELLO(master_name=MASTER_NAME, bot_name=BOT_NAME)
        )

        giphy = GiphyClient()
        giphy.search("Hello!")


    def start_session(self, init: bool = False, nap: bool = False):
        self.worker.stop(init=init)
        self.worker.run(init=init)

        try:
            # Start RTM
            endpoint = self.slackbot.start_real_time_messaging_session()
            listener = MsgListener()
            self.logger.info("start real time messaging session!")

            async def execute_bot():
                ws = await websockets.connect(endpoint)
                while True:
                    receive_json = await ws.recv()
                    listener.handle(receive_json)

            loop = asyncio.new_event_loop()
            asyncio.set_event_loop(loop)
            asyncio.get_event_loop().run_until_complete(execute_bot())
            asyncio.get_event_loop().run_forever()

            if nap:
                self.slackbot.send_message(text=MsgResource.NAP)

        except BaseException:
            self.logger.error(f"Session Error. restart in {self.error_delay} seconds..")
            self.logger.exception("bot")
            time.sleep(5)
            self.start_session(nap=True)

            if self.error_delay <= 1000:
                self.error_delay *= 2
Esempio n. 4
0
class KinoBot:
    def __init__(self) -> None:
        self.slackbot = SlackerAdapter()
        self.logger = Logger().get_logger()
        self.worker = Worker(slackbot=self.slackbot)

        self.error_delay = 1  # Unit (Second)

        # Send a message to channel (init)
        MASTER_NAME = Config.bot.MASTER_NAME
        BOT_NAME = Config.bot.BOT_NAME
        self.slackbot.send_message(
            text=MsgResource.HELLO(master_name=MASTER_NAME, bot_name=BOT_NAME))

        giphy = GiphyClient()
        giphy.search("Hello!")

    def start_session(self, init: bool = False, nap: bool = False):
        self.worker.stop(init=init)
        self.worker.run(init=init)

        try:
            # Start RTM
            endpoint = self.slackbot.start_real_time_messaging_session()
            listener = MsgListener()
            self.logger.info("start real time messaging session!")

            async def execute_bot():
                ws = await websockets.connect(endpoint)
                while True:
                    receive_json = await ws.recv()
                    listener.handle(receive_json)

            loop = asyncio.new_event_loop()
            asyncio.set_event_loop(loop)
            asyncio.get_event_loop().run_until_complete(execute_bot())
            asyncio.get_event_loop().run_forever()

            if nap:
                self.slackbot.send_message(text=MsgResource.NAP)

        except BaseException:
            self.logger.error(
                f"Session Error. restart in {self.error_delay} seconds..")
            self.logger.exception("bot")
            time.sleep(5)
            self.start_session(nap=True)

            if self.error_delay <= 1000:
                self.error_delay *= 2
    def __init__(self) -> None:
        self.slackbot = SlackerAdapter()
        self.logger = Logger().get_logger()
        self.worker = Worker(slackbot=self.slackbot)

        self.error_delay = 5  # Unit (Second)

        # Send a message to channel (init)
        MASTER_NAME = Config.bot.MASTER_NAME
        BOT_NAME = Config.bot.BOT_NAME
        self.slackbot.send_message(
            text=MsgResource.HELLO(master_name=MASTER_NAME, bot_name=BOT_NAME))

        giphy = GiphyClient()
        giphy.search("Hello!")
Esempio n. 6
0
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()
        self.column_list = [
            "attention",
            "productive",
            "happy",
            "sleep",
            "repeat_task",
            "total",
        ]

        if Config.profile.personal:
            from ..utils.profile import Profile

            self.profile = Profile()

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot
Esempio n. 7
0
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()

        self.username = Config.open_api.github.USERNAME
        password = Config.open_api.github.PASSWORD
        self.github = Github(self.username, password)

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general")
            )
        else:
            self.slackbot = slackbot
Esempio n. 8
0
    def __init__(self, slackbot=None):
        self.api = fitbit.api.Fitbit(
            Config.open_api.fitbit.CLIENT_ID,
            Config.open_api.fitbit.CLIENT_SECRET,
            access_token=Config.open_api.fitbit.ACCESS_TOKEN,
            refresh_token="<refresh>",
        )
        self.data_handelr = DataHandler()

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot
Esempio n. 9
0
    def __init__(self) -> None:
        self.slackbot = SlackerAdapter()
        self.logger = Logger().get_logger()
        self.worker = Worker(slackbot=self.slackbot)

        self.error_delay = 1  # Unit (Second)

        # Send a message to channel (init)
        MASTER_NAME = Config.bot.MASTER_NAME
        BOT_NAME = Config.bot.BOT_NAME
        self.slackbot.send_message(
            text=MsgResource.HELLO(master_name=MASTER_NAME, bot_name=BOT_NAME)
        )

        giphy = GiphyClient()
        giphy.search("Hello!")
Esempio n. 10
0
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()
        self.column_list = [
            "attention",
            "productive",
            "happy",
            "sleep",
            "repeat_task",
            "total",
        ]

        if Config.profile.personal:
            from ..utils.profile import Profile

            self.profile = Profile()

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general")
            )
        else:
            self.slackbot = slackbot
Esempio n. 11
0
class GithubManager(object):
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()

        self.username = Config.open_api.github.USERNAME
        password = Config.open_api.github.PASSWORD
        self.github = Github(self.username, password)

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot

    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 __time_point(self, days):
        today = arrow.now()
        point_date = today.replace(days=days)
        point_date = datetime.datetime(point_date.year, point_date.month,
                                       point_date.day)
        return point_date - datetime.timedelta(hours=9)

    def __get_event_count(self, events, start, end):
        commit_events = []
        for event in events:
            if event.created_at > end:
                continue
            if start < event.created_at < end:
                if event.type in ["PushEvent", "PullRequestEvent"]:
                    commit_events.append(event)
            else:
                break
        return len(commit_events)

    def get_point(self):
        commit_count = sum(self.commit(timely="ten_days"))
        return Score.percent(commit_count, 100, 10)
Esempio n. 12
0
class Summary(object):
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()
        self.column_list = [
            "attention",
            "productive",
            "happy",
            "sleep",
            "repeat_task",
            "total",
        ]

        if Config.profile.personal:
            from ..utils.profile import Profile

            self.profile = Profile()

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general"))
        else:
            self.slackbot = slackbot

    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 __get_total_score(self, days="today"):
        if days == "today":
            attention = self.__attention_score()
            productive = self.__productive_score()
            happy = self.__happy_score()
            sleep = self.__sleep_score()
            repeat = self.__repeat_task_score()

            today_data = self.data_handler.read_record()
            summary_data = today_data.get("summary", {})

            do_diary = summary_data.get("do_diary", False)
            do_exercise = summary_data.get("do_exercise", False)
            do_bat = summary_data.get("do_bat", False)

            total = (Score.percent(attention, Config.score.get(
                "ATTENTION", 20), 100) +
                     Score.percent(happy, Config.score.get("HAPPY", 10), 100) +
                     Score.percent(productive,
                                   Config.score.get("PRODUCTIVE", 30), 100) +
                     Score.percent(sleep, Config.score.get("SLEEP", 20), 100) +
                     Score.percent(repeat, Config.score.get("REPEAT_TASK", 10),
                                   100))

            if do_diary:
                total += Config.score.get("DIARY", 5)
            if do_exercise:
                total += Config.score.get("EXERCISE", 5)
            if do_bat:
                total += Config.score.get("BAT", 5)

            if total > 100:
                total = 100

            data = {
                "attention": round(attention * 100) / 100,
                "happy": round(happy * 100) / 100,
                "productive": round(productive * 100) / 100,
                "sleep": round(sleep * 100) / 100,
                "repeat_task": round(repeat * 100) / 100,
                "do_diary": do_diary,
                "do_exercise": do_exercise,
                "do_bat": do_bat,
                "total": round(total * 100) / 100,
            }
            return data
        elif isinstance(days, int):
            data = self.data_handler.read_summary(days=days)
            for c in self.column_list:
                if c not in data:
                    data[c] = 0
            return data

    def __productive_score(self):
        rescue_time_point = RescueTime().get_point()
        toggl_point = TogglManager().get_point()
        github_point = GithubManager().get_point()
        todoist_point = TodoistManager().get_point()

        data = {
            "productive_details": {
                "rescue_time": round(rescue_time_point * 100) / 100,
                "toggl": round(toggl_point * 100) / 100,
                "github": round(github_point * 100) / 100,
                "todoist": round(todoist_point * 100) / 100,
            }
        }
        self.data_handler.edit_summary(data)

        base_point = 0
        rescue_time_ratio = Config.score.productives.get("RESCUE_TIME", 10)
        github_ratio = Config.score.productives.get("GITHUB", 10)
        todoist_ratio = Config.score.productives.get("TODOIST", 50)
        toggl_ratio = Config.score.productives.get("TOGGL", 30)

        if self.is_holiday():
            base_point = Config.score.get("HOLIDAY_BASE", 50)
            holiday_ratio = Config.score.get("HOLIDAY_RATIO", 0.5)

            rescue_time_ratio *= holiday_ratio
            github_ratio *= holiday_ratio
            todoist_ratio *= holiday_ratio
            toggl_ratio *= holiday_ratio

        rescue_time_point = Score.percent(rescue_time_point, rescue_time_ratio,
                                          100)
        github_point = Score.percent(github_point, github_ratio, 100)
        todoist_point = Score.percent(todoist_point, todoist_ratio, 100)
        toggl_point = Score.percent(toggl_point, toggl_ratio, 100)
        return (base_point + rescue_time_point + github_point + todoist_point +
                toggl_point)

    def __attention_score(self):
        BASE_SCORE = 60
        SCORE_UNIT = 8

        record_data = self.data_handler.read_record()
        activity_data = record_data["activity"]
        task_data = activity_data.get("task", [])
        if len(task_data) > 0:
            attention_scores = [
                BASE_SCORE + t.get("score", 3) * SCORE_UNIT for t in task_data
            ]
            return sum(attention_scores) / len(attention_scores)
        else:
            return BASE_SCORE

    def __happy_score(self):
        BASE_SCORE = 60
        SCORE_UNIT = 8

        record_data = self.data_handler.read_record()
        activity_data = record_data["activity"]
        happy_data = activity_data.get("happy", [])
        if len(happy_data) > 0:
            happy_scores = [
                BASE_SCORE + h.get("score", 3) * SCORE_UNIT for h in happy_data
            ]
            return sum(happy_scores) / len(happy_scores)
        else:
            return BASE_SCORE

    def __sleep_score(self):
        self.__get_sleep_time_with_fitbit()

        activity_data = self.data_handler.read_record().get("activity", {})
        sleep_data = activity_data.get("sleep", [])

        if len(sleep_data) == 0:
            sleep_start_time = arrow.now()
            sleep_end_time = arrow.now()
        else:
            for s in sleep_data:
                if s["is_main"]:
                    sleep_start_time = arrow.get(s.get("start_time", None))
                    sleep_end_time = arrow.get(s.get("end_time", None))

        sleep_time = (sleep_end_time - sleep_start_time).seconds / 60 / 60
        sleep_time = sleep_time * 100

        if sleep_time > 800:
            sleep_time -= sleep_time - 800

        if sleep_time > 700:
            sleep_time = 700

        return Score.percent(sleep_time, 100, 700)

    def __get_sleep_time_with_fitbit(self):
        fitbit = Fitbit()
        sleep_data = fitbit.get_sleeps()
        self.data_handler.edit_activity("sleep", sleep_data)

    def __repeat_task_score(self):
        trello = TrelloManager()
        minus_point = 15
        if self.is_holiday():
            minus_point /= 2
        return 100 - (minus_point *
                      trello.get_card_count_by_list_name("Tasks"))

    def record_write_diary(self):
        self.data_handler.edit_summary({"do_diary": True})

    def record_exercise(self):
        self.data_handler.edit_summary({"do_exercise": True})

    def record_bat(self):
        self.data_handler.edit_summary({"do_bat": True})

    def record_holiday(self, dnd):
        self.data_handler.edit_summary({"is_holiday": dnd})

    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 check_sleep_time(self):
        self.slackbot.send_message(text=MsgResource.GOOD_MORNING)

        record = self.data_handler.read_record()
        activity = record.get("activity", {})
        go_to_bed_time = arrow.get(activity.get("go_to_bed", None))

        wake_up_time = arrow.now()
        self.data_handler.edit_record_with_category(
            "activity", ("wake_up", str(wake_up_time)))

        sleep_time = (wake_up_time - go_to_bed_time).seconds / 60 / 60
        sleep_time = round(sleep_time * 100) / 100

        self.data_handler.edit_record(("Sleep", str(sleep_time)))

        self.slackbot.send_message(text=MsgResource.SLEEP_TIME(
            bed_time=go_to_bed_time.format("HH:mm"),
            wakeup_time=wake_up_time.format("HH:mm"),
            diff_h=str(sleep_time),
        ))

    def check_go_to_bed(self):
        go_to_bed_time = arrow.now()
        self.data_handler.edit_record_with_category(
            "activity", ("go_to_bed", str(go_to_bed_time)))

        self.slackbot.send_message(text=MsgResource.GOOD_NIGHT)

        # slack presence issue
        # state = State()
        # state.check()
        # presence_log = state.current[state.SLEEP]
        # if presence_log['presence'] == 'away':
        # go_to_bed_time = arrow.get(presence_log['time'])
        # self.data_handler.edit_record_with_category(
        # 'activity', ('go_to_bed', str(go_to_bed_time)))

    def check_commit_count(self):
        github = GithubManager()
        commit_count = github.commit(timely=-1)
        self.data_handler.edit_summary(("github_commit_count", commit_count))

    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)
Esempio n. 13
0
class Summary(object):
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()
        self.column_list = [
            "attention",
            "productive",
            "happy",
            "sleep",
            "repeat_task",
            "total",
        ]

        if Config.profile.personal:
            from ..utils.profile import Profile

            self.profile = Profile()

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general")
            )
        else:
            self.slackbot = slackbot

    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 __get_total_score(self, days="today"):
        if days == "today":
            attention = self.__attention_score()
            productive = self.__productive_score()
            happy = self.__happy_score()
            sleep = self.__sleep_score()
            repeat = self.__repeat_task_score()

            today_data = self.data_handler.read_record()
            summary_data = today_data.get("summary", {})

            do_diary = summary_data.get("do_diary", False)
            do_exercise = summary_data.get("do_exercise", False)
            do_bat = summary_data.get("do_bat", False)

            total = (
                Score.percent(attention, Config.score.get("ATTENTION", 20), 100)
                + Score.percent(happy, Config.score.get("HAPPY", 10), 100)
                + Score.percent(productive, Config.score.get("PRODUCTIVE", 30), 100)
                + Score.percent(sleep, Config.score.get("SLEEP", 20), 100)
                + Score.percent(repeat, Config.score.get("REPEAT_TASK", 10), 100)
            )

            if do_diary:
                total += Config.score.get("DIARY", 5)
            if do_exercise:
                total += Config.score.get("EXERCISE", 5)
            if do_bat:
                total += Config.score.get("BAT", 5)

            if total > 100:
                total = 100

            data = {
                "attention": round(attention * 100) / 100,
                "happy": round(happy * 100) / 100,
                "productive": round(productive * 100) / 100,
                "sleep": round(sleep * 100) / 100,
                "repeat_task": round(repeat * 100) / 100,
                "do_diary": do_diary,
                "do_exercise": do_exercise,
                "do_bat": do_bat,
                "total": round(total * 100) / 100,
            }
            return data
        elif isinstance(days, int):
            data = self.data_handler.read_summary(days=days)
            for c in self.column_list:
                if c not in data:
                    data[c] = 0
            return data

    def __productive_score(self):
        rescue_time_point = RescueTime().get_point()
        toggl_point = TogglManager().get_point()
        github_point = GithubManager().get_point()
        todoist_point = TodoistManager().get_point()

        data = {
            "productive_details": {
                "rescue_time": round(rescue_time_point * 100) / 100,
                "toggl": round(toggl_point * 100) / 100,
                "github": round(github_point * 100) / 100,
                "todoist": round(todoist_point * 100) / 100,
            }
        }
        self.data_handler.edit_summary(data)

        base_point = 0
        rescue_time_ratio = Config.score.productives.get("RESCUE_TIME", 10)
        github_ratio = Config.score.productives.get("GITHUB", 10)
        todoist_ratio = Config.score.productives.get("TODOIST", 50)
        toggl_ratio = Config.score.productives.get("TOGGL", 30)

        if self.is_holiday():
            base_point = Config.score.get("HOLIDAY_BASE", 50)
            holiday_ratio = Config.score.get("HOLIDAY_RATIO", 0.5)

            rescue_time_ratio *= holiday_ratio
            github_ratio *= holiday_ratio
            todoist_ratio *= holiday_ratio
            toggl_ratio *= holiday_ratio

        rescue_time_point = Score.percent(rescue_time_point, rescue_time_ratio, 100)
        github_point = Score.percent(github_point, github_ratio, 100)
        todoist_point = Score.percent(todoist_point, todoist_ratio, 100)
        toggl_point = Score.percent(toggl_point, toggl_ratio, 100)
        return (
            base_point + rescue_time_point + github_point + todoist_point + toggl_point
        )

    def __attention_score(self):
        BASE_SCORE = 60
        SCORE_UNIT = 8

        record_data = self.data_handler.read_record()
        activity_data = record_data["activity"]
        task_data = activity_data.get("task", [])
        if len(task_data) > 0:
            attention_scores = [BASE_SCORE + t.get("score", 3) * SCORE_UNIT for t in task_data]
            return sum(attention_scores) / len(attention_scores)
        else:
            return BASE_SCORE

    def __happy_score(self):
        BASE_SCORE = 60
        SCORE_UNIT = 8

        record_data = self.data_handler.read_record()
        activity_data = record_data["activity"]
        happy_data = activity_data.get("happy", [])
        if len(happy_data) > 0:
            happy_scores = [BASE_SCORE + h.get("score", 3) * SCORE_UNIT for h in happy_data]
            return sum(happy_scores) / len(happy_scores)
        else:
            return BASE_SCORE

    def __sleep_score(self):
        self.__get_sleep_time_with_fitbit()

        activity_data = self.data_handler.read_record().get("activity", {})
        sleep_data = activity_data.get("sleep", [])

        if len(sleep_data) == 0:
            sleep_start_time = arrow.now()
            sleep_end_time = arrow.now()
        else:
            for s in sleep_data:
                if s["is_main"]:
                    sleep_start_time = arrow.get(s.get("start_time", None))
                    sleep_end_time = arrow.get(s.get("end_time", None))

        sleep_time = (sleep_end_time - sleep_start_time).seconds / 60 / 60
        sleep_time = sleep_time * 100

        if sleep_time > 800:
            sleep_time -= sleep_time - 800

        if sleep_time > 700:
            sleep_time = 700

        return Score.percent(sleep_time, 100, 700)

    def __get_sleep_time_with_fitbit(self):
        fitbit = Fitbit()
        sleep_data = fitbit.get_sleeps()
        self.data_handler.edit_activity("sleep", sleep_data)

    def __repeat_task_score(self):
        trello = TrelloManager()
        minus_point = 15
        if self.is_holiday():
            minus_point /= 2
        return 100 - (minus_point * trello.get_card_count_by_list_name("Tasks"))

    def record_write_diary(self):
        self.data_handler.edit_summary({"do_diary": True})

    def record_exercise(self):
        self.data_handler.edit_summary({"do_exercise": True})

    def record_bat(self):
        self.data_handler.edit_summary({"do_bat": True})

    def record_holiday(self, dnd):
        self.data_handler.edit_summary({"is_holiday": dnd})

    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 check_sleep_time(self):
        self.slackbot.send_message(text=MsgResource.GOOD_MORNING)

        record = self.data_handler.read_record()
        activity = record.get("activity", {})
        go_to_bed_time = arrow.get(activity.get("go_to_bed", None))

        wake_up_time = arrow.now()
        self.data_handler.edit_record_with_category(
            "activity", ("wake_up", str(wake_up_time))
        )

        sleep_time = (wake_up_time - go_to_bed_time).seconds / 60 / 60
        sleep_time = round(sleep_time * 100) / 100

        self.data_handler.edit_record(("Sleep", str(sleep_time)))

        self.slackbot.send_message(
            text=MsgResource.SLEEP_TIME(
                bed_time=go_to_bed_time.format("HH:mm"),
                wakeup_time=wake_up_time.format("HH:mm"),
                diff_h=str(sleep_time),
            )
        )

    def check_go_to_bed(self):
        go_to_bed_time = arrow.now()
        self.data_handler.edit_record_with_category(
            "activity", ("go_to_bed", str(go_to_bed_time))
        )

        self.slackbot.send_message(text=MsgResource.GOOD_NIGHT)

        # slack presence issue
        # state = State()
        # state.check()
        # presence_log = state.current[state.SLEEP]
        # if presence_log['presence'] == 'away':
        # go_to_bed_time = arrow.get(presence_log['time'])
        # self.data_handler.edit_record_with_category(
        # 'activity', ('go_to_bed', str(go_to_bed_time)))

    def check_commit_count(self):
        github = GithubManager()
        commit_count = github.commit(timely=-1)
        self.data_handler.edit_summary(("github_commit_count", commit_count))

    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)
Esempio n. 14
0
class GithubManager(object):
    def __init__(self, slackbot=None):
        self.data_handler = DataHandler()

        self.username = Config.open_api.github.USERNAME
        password = Config.open_api.github.PASSWORD
        self.github = Github(self.username, password)

        if slackbot is None:
            self.slackbot = SlackerAdapter(
                channel=Config.slack.channel.get("REPORT", "#general")
            )
        else:
            self.slackbot = slackbot

    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 __time_point(self, days):
        today = arrow.now()
        point_date = today.replace(days=days)
        point_date = datetime.datetime(
            point_date.year, point_date.month, point_date.day
        )
        return point_date - datetime.timedelta(hours=9)

    def __get_event_count(self, events, start, end):
        commit_events = []
        for event in events:
            if event.created_at > end:
                continue
            if start < event.created_at < end:
                if event.type in ["PushEvent", "PullRequestEvent"]:
                    commit_events.append(event)
            else:
                break
        return len(commit_events)

    def get_point(self):
        commit_count = sum(self.commit(timely="ten_days"))
        return Score.percent(commit_count, 100, 10)