feat:GoogleCalendarにイベントを追加するコマンド

.env-example

@@ -1 +1,2 @@
 DISCORD_BOT_TOKEN=xxxxxxxx
+CALENDAR_WEBHOOK_URL=xxxxxxxx

.gitignore

@@ -138,4 +138,5 @@ dmypy.json
 cython_debug/
 
 # Custom
-.DS_Store
+.DS_Store
+key.json

Pipfile

@@ -10,6 +10,8 @@ black = "*"
 
 [packages]
 "discord.py" = "*"
+google-api-python-client = "*"
+google-auth = "*"
 
 [requires]
 python_version = "3.8"

bot.py

@@ -5,9 +5,7 @@
 
 # bot = commands.Bot(command_prefix="!")
 token = getenv("DISCORD_BOT_TOKEN")
-INITIAL_PLUGINS = [
-    "plugin.echo"
-]
+INITIAL_PLUGINS = ["plugin.echo", "plugin.calendar"]
 
 # bot.run(token)
 class TheGeneralBot(commands.Bot):
@@ -18,7 +16,9 @@ def __init__(self, **kwargs):
                 self.load_extension(cog)
                 print(f"loaded {cog}")
             except Exception as exc:
-                print(f"Could not load extension {cog} due to {exc.__class__.__name__}: {exc}")
+                print(
+                    f"Could not load extension {cog} due to {exc.__class__.__name__}: {exc}"
+                )
 
 
 bot = TheGeneralBot()

logic/calendar.py

@@ -0,0 +1,88 @@
+import datetime
+from abc import ABCMeta, abstractmethod
+from dataclasses import dataclass
+from os import getenv
+
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+
+
+@dataclass
+class Event:
+    title: str
+    date: str
+    starttime: str
+    endtime: str
+    location: str
+
+    def _get_datetime(date, time) -> str:
+        today = datetime.datetime.today()
+        splitteddate = date.split("/")
+        splittedtime = time.split(":")
+
+        return datetime.datetime(
+            today.year,
+            int(splitteddate[0]),
+            int(splitteddate[1]),
+            int(splittedtime[0]),
+            int(splittedtime[1]),
+        ).isoformat()
+
+    @property
+    def get_starttime(self) -> str:
+        datetime = Event._get_datetime(self.date, self.starttime)
+
+        return datetime
+
+    @property
+    def get_endtime(self) -> str:
+        datetime = Event._get_datetime(self.date, self.endtime)
+
+        return datetime
+
+
+class ICalendar(metaclass=ABCMeta):
+    @abstractmethod
+    def _get_service():
+        pass
+
+    @abstractmethod
+    def _create_event_obj(newevent: Event) -> bytes:
+        pass
+
+    @classmethod
+    @abstractmethod
+    def add_event(cls, newevent: Event) -> None:
+        pass
+
+
+class GoogleCalendarImpl(ICalendar):
+    def _get_service():
+        credentials = service_account.Credentials.from_service_account_file("key.json")
+        scoped_credentials = credentials.with_scopes(
+            ["https://www.googleapis.com/auth/calendar"]
+        )
+        service = build("calendar", "v3", credentials=scoped_credentials)
+
+        return service
+
+    def _create_event_obj(self, newevent) -> dict:
+        body = {
+            "summary": newevent.title,
+            "location": newevent.location,
+            "start": {"dateTime": newevent.get_starttime, "timeZone": "Japan",},
+            "end": {"dateTime": newevent.get_endtime, "timeZone": "Japan"},
+        }
+
+        return body
+
+    @classmethod
+    def add_event(cls, newevent) -> None:
+        body = GoogleCalendarImpl._create_event_obj(newevent)
+        service = GoogleCalendarImpl._get_service()
+
+        event = (
+            service.events()
+            .insert(calendarId=getenv("CALENDAR_ID"), body=body)
+            .execute()
+        )

plugin/calendar.py

@@ -0,0 +1,27 @@
+from discord.ext.commands import Bot, Cog, Context, command
+
+from logic.calendar import GoogleCalendarImpl, Event
+
+
+class Calendar(Cog):
+    def __init__(self, bot: Bot) -> None:
+        self.bot = bot
+
+    @command()
+    async def add_event(
+        self,
+        ctx: Context,
+        title: str,
+        date: str,
+        starttime: str,
+        endtime: str,
+        location: str,
+    ) -> None:
+        calendar = GoogleCalendarImpl()
+        newevent = Event(title, date, starttime, endtime, location)
+        calendar.add_event(newevent)
+        await ctx.send("Success!")
+
+
+def setup(bot):
+    bot.add_cog(Calendar(bot))

requirements.txt

@@ -4,6 +4,8 @@ async-timeout==3.0.1; python_full_version >= '3.5.3'
 attrs==19.3.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
 chardet==3.0.4
 discord.py==1.3.4
+google-api-python-client==1.10.0
+google-auth==1.20.1
 idna==2.10; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
 multidict==4.7.6; python_version >= '3.5'
 typing-extensions==3.7.4.2

tests/test_calendar.py

@@ -0,0 +1,16 @@
+import json
+
+from logic.calendar import GoogleCalendarImpl, Event
+
+
+def test_command_parse():
+    calendar = GoogleCalendarImpl()
+    newevent = Event("testtitle", "10/10", "10:00", "11:00", "japan")
+    result = calendar._create_event_obj(newevent)
+    obj = {
+        "summary": "testtitle",
+        "location": "japan",
+        "start": {"dateTime": "2020-10-10T10:00:00", "timeZone": "Japan"},
+        "end": {"dateTime": "2020-10-10T11:00:00", "timeZone": "Japan"},
+    }
+    assert result == obj