Merge pull request #6 from pigeon-sable/develop

Develop

Dockerfile

@@ -1,18 +1,10 @@
 # syntax=docker/dockerfile:1
 
-FROM python:3.11.3 as builder
+FROM python:3.11.4
 WORKDIR /app
-COPY ./requirements.txt /app
-COPY ./src /app
-RUN apt-get update && apt-get install -y vim
+RUN apt-get update && apt-get upgrade -y && apt-get install -y vim && apt-get autoremove -y
 RUN pip install --no-cache-dir --upgrade pip
+ADD ./requirements.txt /app
 RUN pip install -r requirements.txt
-CMD [ "bash" ]
-
-FROM python:3.11.3-alpine3.17
-WORKDIR /app
-COPY ./requirements.txt /app
-COPY ./src /app
-RUN pip install --no-cache-dir --upgrade pip
-RUN pip install -r requirements.txt
-CMD ["python", "vuln_info_bot.py"]
+ADD ./vuln_info_bot /app/vuln_info_bot
+CMD [ "python", "vuln_info_bot/main.py" ]

compose.yaml

@@ -0,0 +1,12 @@
+version: "3.8"
+services:
+  app:
+    build:
+      context: ./
+      dockerfile: Dockerfile
+    env_file:
+      - .env
+    environment:
+      - TZ=Asia/Tokyo
+    volumes:
+      - ./vuln_info_bot:/app/vuln_info_bot

vuln_info_bot/main.py

@@ -20,16 +20,15 @@
 """
 
 __author__ = 'pigeon-sable'
-__version__ = '1.0.1'
-__date__ = '2023/05/14 (Created: 2023/04/19)'
+__version__ = '1.0.2'
+__date__ = '2023/06/02 (Created: 2023/04/19)'
 
 import sys
 
 import discord
 from dotenv import load_dotenv
 
-from web import scraping
-from discord import notify_info
+import notify_info
 
 
 def main() -> int:
@@ -40,14 +39,11 @@ def main() -> int:
 
     load_dotenv()  # .envファイルから環境変数を読み込む
 
-    table_today_vulnerabilities = scraping.table_of_jvn_info(
-        'https://jvndb.jvn.jp/index.html')
-
     client = discord.Client(intents=discord.Intents.default(),
                             messages=discord.Intents.messages,
                             activity=discord.Game(name="/vuln"))
 
-    notify_info.event_method(client, table_today_vulnerabilities)
+    notify_info.event_method(client)
 
     return 0
 

vuln_info_bot/discord/notify_info.py → vuln_info_bot/notify_info.py

@@ -20,8 +20,8 @@
 """
 
 __author__ = 'pigeon-sable'
-__version__ = '1.0.1'
-__date__ = '2023/05/14 (Created: 2023/05/14)'
+__version__ = '1.0.2'
+__date__ = '2023/06/02 (Created: 2023/05/14)'
 
 import datetime
 import os
@@ -30,8 +30,13 @@
 import discord
 from discord.ext import tasks
 
+import scraping
 
-def event_method(client: discord.Client, table_today_vulnerabilities: tuple) -> None:
+GET_TIME = '00:00'
+NOTIFY_TIME = '12:00'
+
+
+def event_method(client: discord.Client) -> None:
     """
     Discord ボットでの処理を書きます。
     イベントが発生したときに、自動的に呼び出されます。
@@ -61,11 +66,11 @@ async def on_ready():
     @tasks.loop(seconds=60)
     async def loop():
         notify_room = client.get_channel(room_id["VULNERABILITY_ROOM_ID"])
-        now = datetime.datetime.now(datetime.timezone(
-            datetime.timedelta(hours=9))).strftime('%H:%M')
-        if now == '00:00':
+        now = datetime.datetime.now().strftime('%H:%M')
+        if now == GET_TIME:
+            table_today_vulnerabilities = scraping.table_of_jvn_info()  # スクレイピングで脆弱性情報を取得する
+        if now == NOTIFY_TIME:
             await notify_room.send('=' * 40)
-            print(now)
             await notify_room.send(f'{datetime.datetime.now().date()} の脆弱性情報をお知らせします。')
             await notify_room.send('-' * 40)
             for summary, hyper_reference, severity in table_today_vulnerabilities:

vuln_info_bot/web/scraping.py → vuln_info_bot/scraping.py

@@ -21,16 +21,17 @@
 
 __author__ = 'pigeon-sable'
 __version__ = '1.0.1'
-__date__ = '2023/05/14 (Created: 2023/05/14)'
+__date__ = '2023/06/02 (Created: 2023/05/14)'
 
 import datetime
 import re
 import requests
+import sys
 
 from bs4 import BeautifulSoup
 
 
-def table_of_jvn_info(the_url_string: str) -> tuple:
+def table_of_jvn_info() -> tuple:
     """
     JVNのWebページをウェブスクレイピングし、サマリーとハイパーリンク、CVSSレベルのタプルを作って応答します。
     抽出できない場合には、Noneを応答します。
@@ -42,6 +43,8 @@ def table_of_jvn_info(the_url_string: str) -> tuple:
         tuple: サマリーとハイパーリンク、CVSSレベルのタプル
     """
 
+    the_url_string = 'https://jvndb.jvn.jp/index.html'
+
     response = requests.get(the_url_string)
     if response.status_code != 200:
         return None
@@ -85,3 +88,17 @@ def table_of_jvn_info(the_url_string: str) -> tuple:
             table.append([summary, hyper_reference, severity])
 
     return table
+
+
+def main() -> int:
+    """
+    ライブラリとして提供する table_of_jvn_info() が正しく動作するかチェックする。
+    """
+    table = table_of_jvn_info('https://jvndb.jvn.jp/index.html')
+    print(table)
+
+    return 0
+
+
+if __name__ == '__main__':  # このスクリプトファイルが直接実行されたときだけ、以下の部分を実行する。
+    sys.exit(main())