feat: implemented twitter feed

This commit is contained in:
Hellow
2024-02-12 22:17:48 +01:00
parent 6e9574f376
commit 75d3d5f1aa
4 changed files with 95 additions and 11 deletions

View File

@@ -1,16 +1,20 @@
import logging
class Feed:
@classmethod
def prompt_auth(cls, existing_config: dict) -> dict:
return existing_config
def __init__(self):
pass
def __init__(self, config: dict = None, **kwargs):
self.logger = logging.getLogger(self.__class__.__name__)
self.config = config or {}
def __enter__(self):
return self
def post(self, message: str):
pass
def post(self, message: str, **kwargs):
print(f"Posting {message}")
def run(self):
self.post("Hello, World!")

View File

@@ -30,8 +30,11 @@ class MastodonFeed(Feed):
access_token=access_token,
)
super().__init__()
super().__init__(**kwargs)
def post(self, message: str):
print(f"Posting to Mastodon: {message}")
kwargs = locals().copy()
self.mastodon.toot(message)
Feed.post(**kwargs)

View File

@@ -0,0 +1,68 @@
from pathlib import Path
from twikit import Client
from twikit.errors import Forbidden, Unauthorized
from . import Feed
from ..utils import Paths, PROGRAM_NAME, prompt
class TwitterFeed(Feed):
CLIENTCRED_PATH: Path = Paths.CONFIG_PATH.joinpath("mastodon_clientcred.secret")
@classmethod
def prompt_auth(cls, existing_config: dict) -> dict:
"""
client.login(
auth_info_1=USERNAME ,
auth_info_2=EMAIL,
password=PASSWORD
)
"""
return {
"auth_info_1": prompt.for_string("Username"),
"auth_info_2": prompt.for_string("Email"),
"password": prompt.for_password("Password"),
}
# https://github.com/d60/twikit
def __init__(self, auth_info_1: str, auth_info_2: str, password: str, cookies: dict = None, **kwargs):
super().__init__(**kwargs)
self.client = Client('en-US')
logged_in = False
cookies = cookies or {}
if cookies is not None:
self.client.http.client.cookies = cookies
try:
self.client.user_id()
logged_in = True
except (Forbidden, Unauthorized):
self.logger.warning("Cookies are expired.")
self.client.http.client.cookies.clear()
if not logged_in:
self.logger.info("Logging in with username and email.")
self.client.login(
auth_info_1=auth_info_1,
auth_info_2=auth_info_2,
password=password,
)
logged_in = True
self.config['cookies'] = dict(self.client.http.client.cookies)
print(self.client.user_id())
print(self.client._base_headers)
def post(self, message: str):
kwargs = locals().copy()
self.client.create_tweet(
text=message,
)
Feed.post(**kwargs)