flibusta-apps/library_updater
1
0
Fork 0
mirror of https://github.com/flibusta-apps/library_updater.git synced 2025-12-06 07:45:35 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add pages to book updater

Browse Source
This commit is contained in:
Bulat Kurbanov
2022-07-21 19:52:10 +03:00
parent 529a2ba4b0
commit 0b24e36ee5
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/app/services/updaters/fl_updater.py
View File

@@ -191,24 +191,25 @@ async def update_fl_books(ctx: dict, *args, prefix: Optional[str] = None, **kwar
row[3], row[3],
row[4], row[4],
row[5] == "1", row[5] == "1",
row[6],
] ]
await postgres.execute( await postgres.execute(
""" """
CREATE OR REPLACE FUNCTION update_book( CREATE OR REPLACE FUNCTION update_book(
source_ smallint, remote_id_ int, title_ varchar, lang_ varchar, source_ smallint, remote_id_ int, title_ varchar, lang_ varchar,
file_type_ varchar, uploaded_ date, is_deleted_ boolean file_type_ varchar, uploaded_ date, is_deleted_ boolean, pages_ int
) RETURNS void AS $$ ) RETURNS void AS $$
BEGIN BEGIN
IF EXISTS (SELECT * FROM books WHERE source = source_ AND remote_id = remote_id_) THEN IF EXISTS (SELECT * FROM books WHERE source = source_ AND remote_id = remote_id_) THEN
UPDATE books SET title = title_, lang = lang_, file_type = file_type_, UPDATE books SET title = title_, lang = lang_, file_type = file_type_,
uploaded = uploaded_, is_deleted = is_deleted uploaded = uploaded_, is_deleted = is_deleted, pages = pages_
WHERE source = source_ AND remote_id = remote_id_; WHERE source = source_ AND remote_id = remote_id_;
RETURN; RETURN;
END IF; END IF;
INSERT INTO books (source, remote_id, title, lang, file_type, uploaded, is_deleted) INSERT INTO books (source, remote_id, title, lang, file_type, uploaded, is_deleted, pages)
VALUES (source_, remote_id_, title_, lang_, file_type_, uploaded_, is_deleted_); VALUES (source_, remote_id_, title_, lang_, file_type_, uploaded_, is_deleted_, pages_);
END; END;
$$ LANGUAGE plpgsql; $$ LANGUAGE plpgsql;
""" """
@@ -216,12 +217,12 @@ async def update_fl_books(ctx: dict, *args, prefix: Optional[str] = None, **kwar
async with mysql.cursor(aiomysql.SSCursor) as cursor: async with mysql.cursor(aiomysql.SSCursor) as cursor:
await cursor.execute( await cursor.execute(
"SELECT BookId, Title, Lang, FileType, Time, Deleted FROM libbook;" "SELECT BookId, Title, Lang, FileType, Time, Deleted, Pages FROM libbook;"
) )
while rows := await cursor.fetchmany(1024): while rows := await cursor.fetchmany(1024):
await postgres.executemany( await postgres.executemany(
"SELECT update_book($1, $2, cast($3 as varchar), cast($4 as varchar), cast($5 as varchar), $6, $7);", "SELECT update_book($1, $2, cast($3 as varchar), cast($4 as varchar), cast($5 as varchar), $6, $7, $8);",
(prepare_book(row) for row in rows), (prepare_book(row) for row in rows),
) )