コミットを比較
5 コミット
539333bd0e
...
d1b1bf2a14
| 作成者 | SHA1 | 日付 | |
|---|---|---|---|
| d1b1bf2a14 | |||
| 5769314b6f | |||
| 27d7cfe972 | |||
| 10c1218caf | |||
| 9835907f8a |
+3
-4
@@ -1,4 +1,3 @@
|
||||
[submodule "ai"]
|
||||
path = ai
|
||||
url = https://git.miteruzo.com/miteruzo/nizika_broadcast
|
||||
branch = main
|
||||
[submodule "nizika_nico"]
|
||||
path = nizika_nico
|
||||
url = https://git.miteruzo.com/miteruzo/nizika_nico
|
||||
|
||||
+49
@@ -0,0 +1,49 @@
|
||||
from datetime import datetime
|
||||
|
||||
from atproto import models
|
||||
|
||||
|
||||
class Client:
|
||||
app: AppNamespace
|
||||
|
||||
def get_current_time_iso (self) -> datetime: ...
|
||||
|
||||
def get_post_thread (
|
||||
self,
|
||||
uri: str,
|
||||
parent_height: int | None = None
|
||||
) -> Response: ...
|
||||
|
||||
def follow (self, did: str) -> None: ...
|
||||
|
||||
|
||||
class AppNamespace:
|
||||
bsky: AppBskyNamespace
|
||||
|
||||
|
||||
class AppBskyNamespace:
|
||||
notification: AppBskyNotificationNamespace
|
||||
|
||||
|
||||
class AppBskyNotificationNamespace:
|
||||
def list_notifications (self) -> Response: ...
|
||||
|
||||
def update_seen (self, seen: dict[str, datetime]) -> None: ...
|
||||
|
||||
|
||||
class Response:
|
||||
notifications: list[Notification]
|
||||
thread: (ThreadViewPost
|
||||
| models.AppBskyFeedDefs.NotFoundPost
|
||||
| models.AppBskyFeedDefs.BlockedPost)
|
||||
|
||||
|
||||
class Notification:
|
||||
is_read: bool
|
||||
reason: str
|
||||
uri: str
|
||||
author: ProfileView
|
||||
|
||||
|
||||
class ProfileView:
|
||||
did: str
|
||||
シンボリックリンク
+1
@@ -0,0 +1 @@
|
||||
db/eloquent.pyi
|
||||
@@ -4,9 +4,12 @@ Bluesky のニジカがいろいろする.
|
||||
"""
|
||||
|
||||
import io
|
||||
import random
|
||||
import sys
|
||||
import time
|
||||
from datetime import datetime, timedelta
|
||||
from datetime import date, datetime
|
||||
from datetime import time as dt_time
|
||||
from datetime import timedelta
|
||||
from typing import cast
|
||||
|
||||
import requests
|
||||
@@ -67,6 +70,12 @@ def main (
|
||||
|
||||
client.login (account.USER_ID, account.PASSWORD)
|
||||
|
||||
got_kiriban_at: date = datetime.now ().date () - timedelta (days = datetime.now ().hour < 15)
|
||||
kiriban_list: list[tuple[int, nico.VideoInfo]] = nico.get_kiriban_list (got_kiriban_at)
|
||||
kiriban_interval: timedelta = ((get_kiriban_dt_to_update () - datetime.now ())
|
||||
/ len (kiriban_list))
|
||||
next_kiriban_at = datetime.now ()
|
||||
|
||||
last_posted_at = datetime.now () - timedelta (hours = 6)
|
||||
has_got_snack_time = False
|
||||
has_taken_hot_spring = False
|
||||
@@ -99,6 +108,36 @@ def main (
|
||||
parent = records[0]['strong_ref'],
|
||||
root = records[-1]['strong_ref']))
|
||||
|
||||
if kiriban_list and datetime.now () >= next_kiriban_at:
|
||||
(views_count, video_info) = (
|
||||
kiriban_list.pop (random.randint (0, len (kiriban_list) - 1)))
|
||||
uri = f"https://www.nicovideo.jp/watch/{ video_info['contentId'] }"
|
||||
(title, description, thumbnail) = get_embed_info (uri)
|
||||
try:
|
||||
upload = client.com.atproto.repo.upload_blob (
|
||||
io.BytesIO (requests.get (thumbnail,
|
||||
timeout = 60).content))
|
||||
thumb = upload.blob
|
||||
except Timeout:
|
||||
thumb = None
|
||||
embed_external = models.AppBskyEmbedExternal.Main (
|
||||
external = models.AppBskyEmbedExternal.External (
|
||||
title = title,
|
||||
description = description,
|
||||
thumb = thumb,
|
||||
uri = uri))
|
||||
client.post (Talk.main (f"""
|
||||
ニコニコの『{ video_info['title'] }』という動画が{ views_count }再生を突破しました。
|
||||
つけられたタグは「{ '」、「'.join (video_info['tags']) }」です。
|
||||
概要には次のように書かれています:
|
||||
```html
|
||||
{ video_info['description'] }
|
||||
```
|
||||
このことについて、ニジカちゃんからのお祝いメッセージを下さい。"""),
|
||||
embed = embed_external)
|
||||
next_kiriban_at += kiriban_interval
|
||||
last_posted_at = now
|
||||
|
||||
latest_deerjika = nico.get_latest_deerjika ()
|
||||
if latest_deerjika is not None:
|
||||
for datum in [e for e in [latest_deerjika]
|
||||
@@ -135,29 +174,37 @@ def main (
|
||||
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 (
|
||||
alt = ('左に喜多ちゃん、右に人面鹿のニジカが'
|
||||
'V字に並んでいる。'
|
||||
'喜多ちゃんは右手でピースサインをして'
|
||||
'片目をウインクしている。'
|
||||
'ニジカは両手を広げ、'
|
||||
'右手にスプーンを持って'
|
||||
'ポーズを取っている。'
|
||||
'背景には'
|
||||
'赤と黄色の放射線状の模様が広がり、'
|
||||
'下部に「おやつタイムだ!!!!」という'
|
||||
'日本語のテキストが表示されている。'),
|
||||
image = client.com.atproto.repo.upload_blob (f).blob)
|
||||
client.post (Talk.main ('おやつタイムだ!!!!'),
|
||||
embed = models.app.bsky.embed.images.Main (
|
||||
images = [image]))
|
||||
last_posted_at = now
|
||||
except Exception:
|
||||
pass
|
||||
has_got_snack_time = True
|
||||
if now.hour == 15:
|
||||
if got_kiriban_at < datetime.now ().date ():
|
||||
kiriban_list = nico.get_kiriban_list (datetime.now ().date ())
|
||||
got_kiriban_at = datetime.now ().date ()
|
||||
kiriban_interval = ((get_kiriban_dt_to_update () - datetime.now ())
|
||||
/ len (kiriban_list))
|
||||
next_kiriban_at = datetime.now ()
|
||||
|
||||
if not has_got_snack_time:
|
||||
try:
|
||||
with open ('./assets/snack-time.jpg', 'rb') as f:
|
||||
image = models.AppBskyEmbedImages.Image (
|
||||
alt = ('左に喜多ちゃん、右に人面鹿のニジカが'
|
||||
'V字に並んでいる。'
|
||||
'喜多ちゃんは右手でピースサインをして'
|
||||
'片目をウインクしている。'
|
||||
'ニジカは両手を広げ、'
|
||||
'右手にスプーンを持って'
|
||||
'ポーズを取っている。'
|
||||
'背景には'
|
||||
'赤と黄色の放射線状の模様が広がり、'
|
||||
'下部に「おやつタイムだ!!!!」という'
|
||||
'日本語のテキストが表示されている。'),
|
||||
image = client.com.atproto.repo.upload_blob (f).blob)
|
||||
client.post (Talk.main ('おやつタイムだ!!!!'),
|
||||
embed = models.app.bsky.embed.images.Main (
|
||||
images = [image]))
|
||||
last_posted_at = now
|
||||
except Exception:
|
||||
pass
|
||||
has_got_snack_time = True
|
||||
|
||||
if now.hour == 20 and has_taken_hot_spring:
|
||||
has_taken_hot_spring = False
|
||||
@@ -192,7 +239,6 @@ def main (
|
||||
time.sleep (60)
|
||||
|
||||
|
||||
|
||||
def get_embed_info (
|
||||
url: str
|
||||
) -> tuple[str, str, str]:
|
||||
@@ -231,5 +277,15 @@ def get_embed_info (
|
||||
return (title, description, thumbnail)
|
||||
|
||||
|
||||
def get_kiriban_dt_to_update (
|
||||
) -> datetime:
|
||||
now = datetime.now ()
|
||||
today = now.date ()
|
||||
dt = datetime.combine (today, dt_time (15, 0))
|
||||
if dt <= now:
|
||||
dt += timedelta (days = 1)
|
||||
return dt
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main (*sys.argv[1:])
|
||||
|
||||
@@ -2,11 +2,32 @@
|
||||
ニコニコのニジカ動画取得モヂュール
|
||||
"""
|
||||
|
||||
from typing import TypedDict
|
||||
import os
|
||||
from datetime import date, timedelta
|
||||
from typing import TypedDict, cast
|
||||
|
||||
import requests
|
||||
from bs4 import BeautifulSoup
|
||||
from requests.exceptions import Timeout
|
||||
from eloquent import DatabaseManager, Model
|
||||
|
||||
from db.models import Tag, Video, VideoHistory, VideoTag
|
||||
|
||||
CONFIG: dict[str, DbConfig] = { 'mysql': { 'driver': 'mysql',
|
||||
'host': 'localhost',
|
||||
'database': 'nizika_nico',
|
||||
'user': os.environ['MYSQL_USER'],
|
||||
'password': os.environ['MYSQL_PASS'],
|
||||
'prefix': '' } }
|
||||
|
||||
DB = DatabaseManager (CONFIG)
|
||||
Model.set_connection_resolver (DB)
|
||||
|
||||
KIRIBAN_VIEWS_COUNTS: set[int] = { *range (100, 1_000, 100),
|
||||
*range (1_000, 10_000, 1_000),
|
||||
*range (10_000, 1_000_001, 10_000),
|
||||
194, 245, 510, 114_514, 1_940, 2_450, 5_100, 24_500,
|
||||
51_000, 2_424 }
|
||||
|
||||
|
||||
class VideoInfo (TypedDict):
|
||||
@@ -38,22 +59,7 @@ def get_latest_deerjika (
|
||||
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
|
||||
return get_video_info (video_info['contentId'])
|
||||
|
||||
|
||||
def get_bs_from_url (
|
||||
@@ -87,3 +93,64 @@ def get_bs_from_url (
|
||||
req.encoding = req.apparent_encoding
|
||||
|
||||
return BeautifulSoup (req.text, 'html.parser')
|
||||
|
||||
|
||||
def get_video_info (
|
||||
video_code: str,
|
||||
) -> VideoInfo | None:
|
||||
video_info: dict[str, str | list[str]] = { 'contentId': video_code }
|
||||
|
||||
bs = get_bs_from_url (f"https://www.nicovideo.jp/watch/{ video_code }")
|
||||
if bs is None:
|
||||
return None
|
||||
|
||||
try:
|
||||
title = bs.find ('title')
|
||||
if title is None:
|
||||
return None
|
||||
video_info['title'] = '-'.join (title.text.split ('-')[:(-1)])[:(-1)]
|
||||
|
||||
tags: str = bs.find ('meta', attrs = { 'name': 'keywords' }).get ('content') # type: ignore
|
||||
video_info['tags'] = tags.split (',')
|
||||
|
||||
video_info['description'] = bs.find ('meta', attrs = { 'name': 'description' }).get ('content') # type: ignore
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
return cast (VideoInfo, video_info)
|
||||
|
||||
|
||||
def get_kiriban_list (
|
||||
base_date: date,
|
||||
) -> list[tuple[int, VideoInfo]]:
|
||||
kiriban_list: list[tuple[int, VideoInfo]] = []
|
||||
|
||||
latest_fetched_at = cast (date, VideoHistory.max ('fetched_at'))
|
||||
previous_fetched_at = cast (date, (VideoHistory
|
||||
.where ('fetched_at', '<', latest_fetched_at)
|
||||
.max ('fetched_at')))
|
||||
|
||||
for kiriban_views_count in KIRIBAN_VIEWS_COUNTS:
|
||||
targets = ({ vh.video.code for vh in (VideoHistory
|
||||
.where ('fetched_at', latest_fetched_at)
|
||||
.where ('views_count', '>=', kiriban_views_count)
|
||||
.get ()) }
|
||||
- { vh.video.code for vh in (VideoHistory
|
||||
.where ('fetched_at', previous_fetched_at)
|
||||
.where ('views_count', '>=', kiriban_views_count)
|
||||
.get ()) })
|
||||
for code in targets:
|
||||
video_info = get_video_info (code)
|
||||
if video_info is not None:
|
||||
kiriban_list.append ((kiriban_views_count, video_info))
|
||||
|
||||
return kiriban_list
|
||||
|
||||
|
||||
class DbConfig (TypedDict):
|
||||
driver: str
|
||||
host: str
|
||||
database: str
|
||||
user: str
|
||||
password: str
|
||||
prefix: str
|
||||
|
||||
サブモジュール
+1
サブモジュール nizika_nico が b2f5f81ca8 で追加されました
新しい課題から参照
ユーザをブロックする