From: Muhammad Rizki <[email protected]>
To: Ammar Faizi <[email protected]>
Cc: Muhammad Rizki <[email protected]>,
GNU/Weeb Mailing List <[email protected]>,
Alviro Iskandar Setiawan <[email protected]>
Subject: [PATCH v5 06/19] Add utility functions for /add_atom
Date: Mon, 1 Aug 2022 07:19:04 +0700 [thread overview]
Message-ID: <[email protected]> (raw)
In-Reply-To: <[email protected]>
Add some utility functions for /add_atom command in the scraper/utils.py
and call it in packages/plugins/add_atom.py
Signed-off-by: Muhammad Rizki <[email protected]>
---
daemon/packages/plugins/manage_atom.py | 20 ++++++++++----------
daemon/scraper/utils.py | 15 +++++++++++++++
2 files changed, 25 insertions(+), 10 deletions(-)
diff --git a/daemon/packages/plugins/manage_atom.py b/daemon/packages/plugins/manage_atom.py
index eca3a60..a10318f 100644
--- a/daemon/packages/plugins/manage_atom.py
+++ b/daemon/packages/plugins/manage_atom.py
@@ -6,17 +6,17 @@
from pyrogram.types import Message
from pyrogram import filters
from packages import DaemonClient
+from scraper import utils
@DaemonClient.on_message(
- filters.command("add_atom") &
- filters.chat(["kiizuah", "nekoha", -1001673279485])
+ filters.command("add_atom") &
+ filters.chat(["kiizuah", "nekoha", -1001673279485])
)
async def add_atom_url(c: DaemonClient, m: Message):
- pass
- # text = utils.remove_command(m.text)
- # if not utils.is_atom_url(text):
- # return
-
- # ###
- # ### TODO: Muhammad Rizki: Add atom url into the database
- # ###
+ text = utils.remove_command(m.text)
+ if not utils.is_atom_url(text):
+ return
+
+ ###
+ ### TODO: Muhammad Rizki: Add atom url into the database
+ ###
diff --git a/daemon/scraper/utils.py b/daemon/scraper/utils.py
index 30efb7b..f7de5ba 100644
--- a/daemon/scraper/utils.py
+++ b/daemon/scraper/utils.py
@@ -12,6 +12,7 @@ import uuid
import os
import re
import shutil
+import httpx
def get_email_msg_id(mail):
@@ -231,3 +232,17 @@ def extract_email_msg_id(msg_id):
return None
return ret.group(1)
+
+async def is_atom_url(text: str):
+ try:
+ async with httpx.AsyncClient() as ses:
+ res = await ses.get(text)
+ mime = res.headers.get("Content-Type")
+
+ return mime == "application/atom+xml"
+ except: return False
+
+def remove_command(text: str):
+ txt = text.split(" ")
+ txt = text.replace(txt[0] + " ","")
+ return txt
--
Muhammad Rizki
next prev parent reply other threads:[~2022-08-01 0:19 UTC|newest]
Thread overview: 26+ messages / expand[flat|nested] mbox.gz Atom feed top
2022-08-01 0:18 [PATCH v5 00/19] New feature to handle atom URLs and broadcast Muhammad Rizki
2022-08-01 0:18 ` [PATCH v5 01/19] Add new table for atom URLs and broadcast chats in db.sql Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 02/19] Add a comment in .gitignore Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 03/19] Add ATOM_URLS and BROADCAST_CHAT methods Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 04/19] Fix MySQL InternalError: Unread result found Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 05/19] Create /add_atom command Muhammad Rizki
2022-08-02 21:43 ` Ammar Faizi
2022-08-01 0:19 ` Muhammad Rizki [this message]
2022-08-01 0:19 ` [PATCH v5 07/19] Refactor database methods Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 08/19] Add insert_atom() in scraper/db.py Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 09/19] Rename admin.py to debugger.py Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 10/19] Add delete_atom() function in scraper/db.py Muhammad Rizki
2022-08-02 21:45 ` Ammar Faizi
2022-08-01 0:19 ` [PATCH v5 11/19] Add utility function buton_numbers() Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 12/19] Move bot commands related files to packages/plugins/commands/ Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 13/19] Add del_atom callback query Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 14/19] Add insert_broadcast() in scraper/db.py Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 15/19] Add delete_broadcast() " Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 16/19] Create /add_bc bot command Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 17/19] Add create_chat_link() function Muhammad Rizki
2022-08-01 0:19 ` [PATCH v5 18/19] Add /del_bc bot command Muhammad Rizki
2022-08-02 21:47 ` Ammar Faizi
2022-08-01 0:19 ` [PATCH v5 19/19] Add del_chat callback query Muhammad Rizki
2022-08-02 21:47 ` Ammar Faizi
2022-08-03 0:13 ` [PATCH v5 00/19] New feature to handle atom URLs and broadcast Ammar Faizi
2022-08-03 0:14 ` Ammar Faizi
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
[email protected] \
[email protected] \
[email protected] \
[email protected] \
[email protected] \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox