リプライが送信されないバグ修正（#17） (#18)

#17

Co-authored-by: miteruzo <miteruzo@naver.com>
Reviewed-on: #18

.gitignore

@@ -1,3 +1,5 @@
 /__pycache__
 /account.py
 /connection.py
+/db
+/eloquent.pyi

.gitmodules

@@ -1,4 +1,6 @@
-[submodule "ai"]
+[submodule "nizika_ai"]
-	path = ai
+	path = nizika_ai
-	url = https://git.miteruzo.com/miteruzo/nizika_broadcast
+	url = https://git.miteruzo.com/miteruzo/nizika_ai.git
-	branch = main
+[submodule "nicolib"]
+	path = nicolib
+	url = https://git.miteruzo.com/miteruzo/nicolib.git

main.py

@@ -1,145 +1,150 @@
-"""
+from __future__ import annotations
-Bluesky のニジカがいろいろする．
-（近々機能ごとにファイル分けて systemd でイベント管理する予定）
-"""
 
-import io
+import asyncio
-import sys
+import os
 import time
-from datetime import datetime, timedelta
+from datetime import datetime
-from typing import cast
+from io import BytesIO
+from typing import Any, TypedDict
 
+import atproto  # type: ignore
 import requests
-from atproto import Client, models
+from atproto import Client  # type: ignore
-from bs4 import BeautifulSoup
+from atproto_client.models import AppBskyEmbedExternal, AppBskyEmbedImages, ComAtprotoRepoStrongRef  # type: ignore
+from atproto_client.models.app.bsky.feed.get_timeline import Response  # type: ignore
+from atproto_client.models.app.bsky.feed.post import ReplyRef  # type: ignore
 from requests.exceptions import Timeout
 
 import account
-import nico
+import nicolib
-from ai.talk import Talk
+from nizika_ai.consts import Character, GPTModel, Platform, QueryType
+from nizika_ai.models import Answer, AnsweredFlag, Query, User
 
+TARGET_WORDS = ['deerjika', 'ニジカ', 'ぼっち', '虹夏', '郁代', 'バーカ',
+                'kfif', 'kita-flatten-ikuyo-flatten', 'ラマ田', 'ゴートう',
+                'ぼざクリ', 'オオミソカ', '伊地知', '喜多ちゃん',
+                '喜タイ', '洗澡鹿', 'シーザオ', '今日は本当に',
+                'ダイソーで', '変なチンチン', 'daisoで', 'だね～(笑)',
+                'おやつタイム', 'わさしが', 'わさび県', 'たぬマ', 'にくまる',
+                'ルイズマリー', '餅', 'ニジゴ', 'ゴニジ', 'ニジニジ',
+                '新年だよね', 'うんこじゃん', 'ほくほくのジャガイモ']
 
-def check_notifications (
-        client: Client,
-) -> list:
-    (uris, last_seen_at) = ([], client.get_current_time_iso ())
-
-    for notification in (client.app.bsky.notification.list_notifications ()
-                         .notifications):
-        if not notification.is_read:
-            if notification.reason in ['mention', 'reply', 'quote']:
-                uris += [notification.uri]
-            elif notification.reason == 'follow':
-                client.follow (notification.author.did)
-
-    client.app.bsky.notification.update_seen ({ 'seen_at': last_seen_at })
-
-    return uris
-
-
-def get_thread_contents (
-        client:         Client,
-        uri:            str,
-        parent_height:  int,
-) -> list:
-    response = (client.get_post_thread (uri           = uri,
-                                        parent_height = parent_height)
-                .thread)
-    records = []
-    while response is not None:
-        records += [{ 'strong_ref': models.create_strong_ref (response.post),
-                      'did':        response.post.author.did,
-                      'handle':     response.post.author.handle,
-                      'name':       response.post.author.display_name,
-                      'datetime':   response.post.record.created_at,
-                      'text':       response.post.record.text,
-                      'embed':      response.post.record.embed }]
-        response = response.parent
-
-    return records
-
-
-def main (
-) -> None:
 time.sleep (60)
 
 client = Client (base_url = 'https://bsky.social')
-
 client.login (account.USER_ID, account.PASSWORD)
 
-    last_posted_at = datetime.now () - timedelta (hours = 6)
+
-    has_got_snack_time = False
+async def main (
-    has_taken_hot_spring = False
+) -> None:
-    watched_videos = []
+    """
+    メーン処理
+    """
+
+    await asyncio.gather (like_posts (),
+                          check_mentions (),
+                          answer ())
+
+
+async def like_posts (
+) -> None:
     while True:
-        now = datetime.now ()
-
-        for uri in check_notifications (client):
-            records = get_thread_contents (client, uri, 20)
-            if len (records) > 0:
-                answer = Talk.main ((records[0]['text']
-                                     if (records[0]['embed'] is None
-                                         or not hasattr (records[0]['embed'],
-                                                         'images'))
-                                     else [
-                        { 'type': 'text', 'text': records[0]['text'] },
-                        { 'type': 'image_url', 'image_url': {
-                            'url': f"https://cdn.bsky.app/img/feed_fullsize/plain/{ records[0]['did'] }/{ records[0]['embed'].images[0].image.ref.link }" } }]),
-                                    records[0]['name'],
-                                    [*map (lambda record: {
-                                               'role': ('assistant'
-                                                        if (record['handle']
-                                                            == account.USER_ID)
-                                                        else 'user'),
-                                               'content':
-                                                   record['text']},
-                                           reversed (records[1:]))])
-                client.post (answer,
-                             reply_to = models.AppBskyFeedPost.ReplyRef (
-                                 parent = records[0]['strong_ref'],
-                                 root   = records[-1]['strong_ref']))
-
-        latest_deerjika = nico.get_latest_deerjika ()
-        if latest_deerjika is not None:
-            for datum in [e for e in [latest_deerjika]
-                            if e['contentId'] not in watched_videos]:
-                watched_videos += [datum['contentId']]
-
-                uri = f"https://www.nicovideo.jp/watch/{ datum['contentId'] }"
-                (title, description, thumbnail) = get_embed_info (uri)
         try:
-                    upload = client.com.atproto.repo.upload_blob (
+            for post in fetch_target_posts ():
-                            io.BytesIO (requests.get (thumbnail,
+                client.like (**post)
-                                                      timeout = 60).content))
+        except Exception as e:
+            print (f"[like_posts] { type (e).__name__ }: { e }")
+
+        await asyncio.sleep (60)
+
+
+async def check_mentions (
+) -> None:
+    while True:
+        try:
+            for uri in check_notifications ():
+                records = fetch_thread_contents (uri, 20)
+                if records:
+                    record = records[0]
+                    image_url: str | None = None
+                    if record['embed'] and hasattr (record['embed'], 'images'):
+                        image_url = ('https://cdn.bsky.app/img/feed_fullsize/plain'
+                                     f"/{ record['did'] }"
+                                     f"/{ record['embed'].images[0].image.ref.link }")
+                    user = _fetch_user (record['did'], record['name'])
+                    _add_query (user, record['text'], image_url, {
+                            'uri':  record['strong_ref']['uri'],
+                            'cid':  record['strong_ref']['cid'] })
+        except Exception as e:
+            print (f"[check_mentions] { type (e).__name__ }: { e }")
+
+        await asyncio.sleep (60)
+
+
+async def answer (
+) -> None:
+    while True:
+        answered_flags = (
+                AnsweredFlag
+                .where ('platform', Platform.BLUESKY.value)
+                .where ('answered', False)
+                .get ())
+        for answered_flag in answered_flags:
+            td: dict[str, Any]
+            answer = answered_flag.answer
+            answered_flag.answered = True
+            answered_flag.save ()
+            match QueryType (answer.query_rel.query_type):
+                case QueryType.BLUESKY_COMMENT:
+                    td = answer.query_rel.transfer_data or { }
+                    uri: str | None = td.get ('uri')
+                    cid: str | None = td.get ('cid')
+                    if (not uri) or (not cid):
+                        continue
+
+                    strong_ref = ComAtprotoRepoStrongRef.Main (uri = uri, cid = cid)
+                    reply_ref = ReplyRef (root = strong_ref, parent = strong_ref)
+                    try:
+                        client.post (answer.content[:250], reply_to = reply_ref)
+                    except Exception as e:
+                        print (f"[answer/reply] { type (e).__name__ }: { e }")
+                        continue
+                case QueryType.KIRIBAN | QueryType.NICO_REPORT:
+                    td = answer.query_rel.transfer_data or { }
+                    video_code: str | None = td.get ('video_code')
+                    if not video_code:
+                        continue
+
+                    uri = f"https://www.nicovideo.jp/watch/{ video_code }"
+                    (title, description, thumbnail) = nicolib.fetch_embed_info (uri)
+                    try:
+                        resp = requests.get (thumbnail, timeout = 60)
+                        resp.raise_for_status ()
+                        upload = client.com.atproto.repo.upload_blob (BytesIO (resp.content))
                         thumb = upload.blob
                     except Timeout:
                         thumb = None
+                    except Exception as e:
+                        print (f"[answer/nico-thumb] { type (e).__name__ }: { e }")
+                        thumb = None
 
-                embed_external = models.AppBskyEmbedExternal.Main (
+                    external = AppBskyEmbedExternal.External (
-                        external = models.AppBskyEmbedExternal.External (
                             title       = title,
                             description = description,
                             thumb       = thumb,
-                            uri         = uri))
+                            uri         = uri)
-                client.post (Talk.main (f"""
+                    embed_external = AppBskyEmbedExternal.Main (external = external)
-ニコニコに『{ datum['title'] }』という動画がアップされました。
+                    try:
-つけられたタグは「{ '」、「'.join (datum['tags']) }」です。
+                        client.post (answer.content[:250], embed = embed_external)
-概要には次のように書かれています：
+                    except Exception as e:
-```html
+                        print (f"[answer/nico-post] { type (e).__name__ }: { e }")
-{ datum['description'] }
+                        continue
-```
+                case QueryType.SNACK_TIME:
-このことについて、みんなに告知するとともに、ニジカちゃんの感想を教えてください。 """),
-                             embed = embed_external)
-                last_posted_at = now
-
-        if now.hour == 14 and has_got_snack_time:
-            has_got_snack_time = False
-
-        if now.hour == 15 and not has_got_snack_time:
                     try:
                         with open ('./assets/snack-time.jpg', 'rb') as f:
-                    image = models.AppBskyEmbedImages.Image (
+                            image = AppBskyEmbedImages.Image (
-                            alt   = ('左に喜多ちゃん、右に人面鹿のニジカが'
+                                    alt = (
+                                        '左に喜多ちゃん、右に人面鹿のニジカが'
                                         'V字に並んでいる。'
                                         '喜多ちゃんは右手でピースサインをして'
                                         '片目をウインクしている。'
@@ -151,21 +156,14 @@ def main (
                                         '下部に「おやつタイムだ!!!!」という'
                                         '日本語のテキストが表示されている。'),
                                     image = client.com.atproto.repo.upload_blob (f).blob)
-                client.post (Talk.main ('おやつタイムだ!!!!'),
+                        client.post (answer.content[:250],
-                             embed = models.app.bsky.embed.images.Main (
+                                     embed = AppBskyEmbedImages.Main (images = [image]))
-                                 images = [image]))
-                last_posted_at = now
                     except Exception:
                         pass
-            has_got_snack_time = True
+                case QueryType.HOT_SPRING:
-
-        if now.hour == 20 and has_taken_hot_spring:
-            has_taken_hot_spring = False
-
-        if now.hour == 21 and not has_taken_hot_spring:
                     try:
                         with open ('./assets/hot-spring.jpg', 'rb') as f:
-                    image = models.AppBskyEmbedImages.Image (
+                            image = AppBskyEmbedImages.Image (
                                     alt = ('左に喜多ちゃん、右にわさび県産滋賀県が'
                                            'V字に並んでいる。'
                                            '喜多ちゃんは右手でピースサインをして'
@@ -177,59 +175,131 @@ def main (
                                            '下部に「温泉に入ろう!!!」という'
                                            '日本語のテキストが表示されている。'),
                                     image = client.com.atproto.repo.upload_blob (f).blob)
-                client.post (Talk.main ('温泉に入ろう!!!'),
+                        client.post (answer.content[:250],
-                             embed = models.app.bsky.embed.images.Main (
+                                     embed = AppBskyEmbedImages.Main (images = [image]))
-                                 images = [image]))
-                last_posted_at = now
                     except Exception:
                         pass
-            has_taken_hot_spring = True
+        await asyncio.sleep (10)
-
-        if now - last_posted_at >= timedelta (hours = 6):
-            client.post (Talk.main ('今どうしてる?'))
-            last_posted_at = now
-
-        time.sleep (60)
 
 
+def check_notifications (
+) -> list[str]:
+    uris: list[str] = []
+    last_seen_at = client.get_current_time_iso ()
 
-def get_embed_info (
+    notifications = client.app.bsky.notification.list_notifications ()
-        url:    str
+    for notification in notifications.notifications:
-) -> tuple[str, str, str]:
+        if not notification.is_read:
-    title:          str = ''
+            match notification.reason:
-    description:    str = ''
+                case 'mention' | 'reply' | 'quote':
-    thumbnail:      str = ''
+                    uris.append (notification.uri)
+                case 'follow':
+                    client.follow (notification.author.did)
 
-    try:
+    client.app.bsky.notification.update_seen ({ 'seen_at': last_seen_at })
-        res = requests.get (url, timeout = 60)
-    except Timeout:
-        return ('', '', '')
 
-    if res.status_code != 200:
+    return uris
-        return ('', '', '')
 
-    soup = BeautifulSoup (res.text, 'html.parser')
 
-    tmp = soup.find ('title')
+def fetch_thread_contents (
-    if tmp is not None:
+        uri:            str,
-        title = tmp.text
+        parent_height:  int,
+) -> list[Record]:
+    post_thread = client.get_post_thread (uri = uri, parent_height = parent_height)
+    if not post_thread:
+        return []
 
-    tmp = soup.find ('meta', attrs = { 'name': 'description' })
+    res = post_thread.thread
-    if tmp is not None and hasattr (tmp, 'get'):
-        try:
-            description = cast (str, tmp.get ('content'))
-        except Exception:
-            pass
 
-    tmp = soup.find ('meta', attrs = { 'name': 'thumbnail' })
+    records: list[Record] = []
-    if tmp is not None and hasattr (tmp, 'get'):
+    while res:
-        try:
+        if hasattr (res, 'post'):
-            thumbnail = cast (str, tmp.get ('content'))
+            records.append ({ 'strong_ref': { 'uri':    res.post.uri,
-        except Exception:
+                                              'cid':    res.post.cid },
-            pass
+                              'did':        res.post.author.did,
+                              'handle':     res.post.author.handle,
+                              'name':       (res.post.author.display_name
+                                             or res.post.author.handle),
+                              'datetime':   res.post.record.created_at,
+                              'text':       getattr (res.post.record, 'text', None) or '',
+                              'embed':      getattr (res.post.record, 'embed', None) })
+            res = res.parent
+        else:
+            break
 
-    return (title, description, thumbnail)
+    return records
+
+
+def fetch_target_posts (
+) -> list[LikeParams]:
+    posts: list[LikeParams] = []
+
+    timeline: Response = client.get_timeline ()
+    for feed in timeline.feed:
+        me = getattr (client, 'me', None)
+        my_did = me.did if me else ''
+        if (feed.post.author.did != my_did
+                and (feed.post.viewer.like is None)
+                and any (target_word in (feed.post.record.text or '').casefold ()
+                         for target_word in TARGET_WORDS)):
+            posts.append (LikeParams ({ 'uri': feed.post.uri, 'cid': feed.post.cid }))
+
+    return posts
+
+
+def _add_query (
+        user:           User,
+        content:        str,
+        image_url:      str | None  = None,
+        transfer_data:  dict[str, Any] | None = None,
+) -> None:
+    query = Query ()
+    query.user_id = user.id
+    query.target_character = Character.DEERJIKA.value
+    query.content = content
+    query.image_url = image_url or None
+    query.query_type = QueryType.BLUESKY_COMMENT.value
+    query.model = GPTModel.GPT3_TURBO.value
+    query.sent_at = datetime.now ()
+    query.answered = False
+    query.transfer_data = transfer_data
+    query.save ()
+    # TODO: 履歴情報の追加
+
+
+def _fetch_user (
+        did:    str,
+        name:   str,
+) -> User:
+    user = User.where ('platform', Platform.BLUESKY.value).where ('code', did).first ()
+    if user is None:
+        user = User ()
+        user.platform = Platform.BLUESKY.value
+        user.code = did
+    user.name = name
+    user.save ()
+    return user
+
+
+class LikeParams (TypedDict):
+    uri:    str
+    cid:    str
+
+
+class Record (TypedDict):
+    strong_ref: StrongRef
+    did:        str
+    handle:     str
+    name:       str
+    datetime:   str
+    text:       str
+    embed:      object
+
+
+class StrongRef (TypedDict):
+    uri:    str
+    cid:    str
 
 
 if __name__ == '__main__':
-    main (*sys.argv[1:])
+    asyncio.run (main ())

nico.py

@@ -1,89 +0,0 @@
-"""
-ニコニコのニジカ動画取得モヂュール
-"""
-
-from typing import TypedDict
-
-import requests
-from bs4 import BeautifulSoup
-from requests.exceptions import Timeout
-
-
-class VideoInfo (TypedDict):
-    contentId:      str
-    title:          str
-    tags:           list[str]
-    description:    str
-
-
-def get_latest_deerjika (
-) -> VideoInfo | None:
-    tag = '伊地知ニジカ OR ぼざろクリーチャーシリーズ'
-    url = f"https://www.nicovideo.jp/tag/{ tag }"
-
-    params = { 'sort':  'f',
-               'order': 'd' }
-
-    video_info = { }
-
-    bs = get_bs_from_url (url, params)
-    if bs is None:
-        return None
-
-    try:
-        video = (bs.find_all ('ul', class_ = 'videoListInner')[1]
-                 .find ('li', class_ = 'item'))
-
-        video_info['contentId'] = video['data-video-id']
-    except Exception:
-        return None
-
-    bs = get_bs_from_url ('https://www.nicovideo.jp/watch/'
-                          + video_info['contentId'])
-    if bs is None:
-        return None
-
-    try:
-        video_info['title'] = '-'.join (bs.find ('title').text.split ('-')[:(-1)])[:(-1)]
-
-        tags = bs.find ('meta', attrs = { 'name': 'keywords' }).get ('content')
-        video_info['tags'] = tags.split (',')
-
-        video_info['description'] = bs.find ('meta', attrs = { 'name': 'description' }).get ('content')
-    except Exception:
-        return None
-
-    return video_info
-
-
-def get_bs_from_url (
-        url:    str,
-        params: dict    = { },
-) -> BeautifulSoup | None:
-    """
-    URL から BeautifulSoup インスタンス生成
-
-    Parameters
-    ----------
-    url:    str
-        捜査する URL
-    params: dict
-        パラメータ
-
-    Return
-    ------
-    BeautifulSoup | None
-        BeautifulSoup オブゼクト（失敗したら None）
-    """
-
-    try:
-        req = requests.get (url, params = params, timeout = 60)
-    except Timeout:
-        return None
-
-    if req.status_code != 200:
-        return None
-
-    req.encoding = req.apparent_encoding
-
-    return BeautifulSoup (req.text, 'html.parser')