mirror of
https://github.com/migatu/conjurer.git
synced 2026-07-15 05:48:35 +00:00
Compare commits
103 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 82bb6cbbad | |||
| e54150fd5d | |||
| 167fe59456 | |||
| 3a02846a17 | |||
| 9c3e51631c | |||
| c3dd4ad8cf | |||
| 5d51e99c70 | |||
| 2c9c86990c | |||
| b87cf7f4d8 | |||
| a68745713b | |||
| bbe1b89007 | |||
| 27fefdccb5 | |||
| 64849f728f | |||
| 9b09c2c13f | |||
| f2fb2232ee | |||
| 7e4dfbf8cd | |||
| 81a387eb4f | |||
| 408f6e5c4e | |||
| a266e5d240 | |||
| 917b57c586 | |||
| 0e15218abd | |||
| f56773b2e1 | |||
| 944dfe2c18 | |||
| 2fd362a1d5 | |||
| 47945ab3dd | |||
| c0b362df6d | |||
| 575237812f | |||
| 60787d8162 | |||
| 8dc4eef149 | |||
| 0f49cb89a3 | |||
| 99f24e6b77 | |||
| 9b14610cc9 | |||
| 14df489b38 | |||
| 8a99ee7c99 | |||
| 3684003150 | |||
| f3b03827e2 | |||
| 848668b8ed | |||
| f1cc24cc06 | |||
| 1cbd44c333 | |||
| 9ba3f1d3ed | |||
| 032cc41c9b | |||
| 217533c91f | |||
| 80f4254ebe | |||
| 5967aefda7 | |||
| 3b232ac3d7 | |||
| 7c0de660fd | |||
| 7de1ad43fd | |||
| fcbda81213 | |||
| 1e00c1fe33 | |||
| 5e810d23ab | |||
| 439c5c753c | |||
| cc11d3f75e | |||
| a35d7da6e3 | |||
| 2c50318f1c | |||
| addc534c36 | |||
| 430b4de995 | |||
| 9d00e40b41 | |||
| 7d29c1d397 | |||
| ce3ad9f8f0 | |||
| abef18f4ad | |||
| 5beb1c6fdd | |||
| ff69091495 | |||
| 69d3c8849c | |||
| b2aab28460 | |||
| 082b25c89c | |||
| 6fefede271 | |||
| c1054241eb | |||
| 04ad1e7005 | |||
| 89bd1b16ae | |||
| 90674c288f | |||
| b4d4f47519 | |||
| 477430e9a3 | |||
| 7dc53ef6b4 | |||
| 837d9be348 | |||
| 499d9c5715 | |||
| 70959da16a | |||
| 8f360417a9 | |||
| aad0477e29 | |||
| fe81f5e8fc | |||
| 6d51412ea3 | |||
| 382cf5d501 | |||
| e121c6dc39 | |||
| 548496e0a6 | |||
| 99409f7c0b | |||
| fd1107cd1b | |||
| f08ce02e85 | |||
| 78346d8448 | |||
| c780308541 | |||
| cc0af0ce32 | |||
| 99e3dd7dca | |||
| 16ead67cf3 | |||
| 5ca73d6ac9 | |||
| f3e690f395 | |||
| 8293b3a69c | |||
| 0ac9006665 | |||
| d912d9bc1b | |||
| b08642dd88 | |||
| 8b1950823b | |||
| 6589b73329 | |||
| 89003c32c9 | |||
| 17eace3e73 | |||
| 3f2248336d | |||
| f8664c0fd1 |
@@ -197,3 +197,7 @@ node_modules/prettier/esm/parser-postcss.mjs
|
|||||||
node_modules/prettier/esm/parser-typescript.mjs
|
node_modules/prettier/esm/parser-typescript.mjs
|
||||||
node_modules/prettier/esm/parser-yaml.mjs
|
node_modules/prettier/esm/parser-yaml.mjs
|
||||||
node_modules/prettier/esm/standalone.mjs
|
node_modules/prettier/esm/standalone.mjs
|
||||||
|
cr_results.json
|
||||||
|
not_in_db.json
|
||||||
|
rr_results.json
|
||||||
|
s_results.json
|
||||||
|
|||||||
@@ -6,8 +6,4 @@
|
|||||||
plugins
|
plugins
|
||||||
user_trunk.yaml
|
user_trunk.yaml
|
||||||
user.yaml
|
user.yaml
|
||||||
<<<<<<< HEAD
|
|
||||||
tmp
|
tmp
|
||||||
=======
|
|
||||||
tools
|
|
||||||
>>>>>>> c7dc465 (Parametryzacja "muzyki mojego ludu")
|
|
||||||
|
|||||||
@@ -0,0 +1,7 @@
|
|||||||
|
enable=all
|
||||||
|
source-path=SCRIPTDIR
|
||||||
|
disable=SC2154
|
||||||
|
|
||||||
|
# If you're having issues with shellcheck following source, disable the errors via:
|
||||||
|
# disable=SC1090
|
||||||
|
# disable=SC1091
|
||||||
+15
-48
@@ -2,72 +2,39 @@
|
|||||||
# To learn more about the format of this file, see https://docs.trunk.io/reference/trunk-yaml
|
# To learn more about the format of this file, see https://docs.trunk.io/reference/trunk-yaml
|
||||||
version: 0.1
|
version: 0.1
|
||||||
cli:
|
cli:
|
||||||
<<<<<<< HEAD
|
version: 1.22.2
|
||||||
version: 1.20.1
|
|
||||||
# Trunk provides extensibility via plugins. (https://docs.trunk.io/plugins)
|
# Trunk provides extensibility via plugins. (https://docs.trunk.io/plugins)
|
||||||
plugins:
|
plugins:
|
||||||
sources:
|
sources:
|
||||||
- id: trunk
|
- id: trunk
|
||||||
ref: v1.4.5
|
ref: v1.6.0
|
||||||
uri: https://github.com/trunk-io/plugins
|
uri: https://github.com/trunk-io/plugins
|
||||||
# Many linters and tools depend on runtimes - configure them here. (https://docs.trunk.io/runtimes)
|
# Many linters and tools depend on runtimes - configure them here. (https://docs.trunk.io/runtimes)
|
||||||
=======
|
|
||||||
version: 1.12.1
|
|
||||||
plugins:
|
|
||||||
sources:
|
|
||||||
- id: trunk
|
|
||||||
ref: v0.0.22
|
|
||||||
uri: https://github.com/trunk-io/plugins
|
|
||||||
lint:
|
|
||||||
enabled:
|
|
||||||
- actionlint@1.6.25
|
|
||||||
- markdownlint@0.35.0
|
|
||||||
- trivy@0.43.1
|
|
||||||
- trufflehog@3.44.0
|
|
||||||
- pylint
|
|
||||||
- codespell
|
|
||||||
- cspell
|
|
||||||
- git-diff-check
|
|
||||||
- eslint
|
|
||||||
- prettier
|
|
||||||
- semgrep
|
|
||||||
- autopep8
|
|
||||||
- bandit
|
|
||||||
- black
|
|
||||||
- flake8
|
|
||||||
- isort
|
|
||||||
- mypy
|
|
||||||
- pylint
|
|
||||||
- ruff
|
|
||||||
- semgrep
|
|
||||||
- yapf
|
|
||||||
- gitleaks
|
|
||||||
|
|
||||||
>>>>>>> c7dc465 (Parametryzacja "muzyki mojego ludu")
|
|
||||||
runtimes:
|
runtimes:
|
||||||
enabled:
|
enabled:
|
||||||
|
- go@1.21.0
|
||||||
- node@18.12.1
|
- node@18.12.1
|
||||||
- python@3.10.8
|
- python@3.10.8
|
||||||
<<<<<<< HEAD
|
|
||||||
# This is the section where you manage your linters. (https://docs.trunk.io/check/configuration)
|
# This is the section where you manage your linters. (https://docs.trunk.io/check/configuration)
|
||||||
lint:
|
lint:
|
||||||
enabled:
|
enabled:
|
||||||
- actionlint@1.6.27
|
- shellcheck@0.10.0
|
||||||
- bandit@1.7.8
|
- shfmt@3.6.0
|
||||||
- black@24.3.0
|
- taplo@0.8.1
|
||||||
|
- actionlint@1.7.1
|
||||||
|
- bandit@1.7.9
|
||||||
|
- black@24.4.2
|
||||||
- codespell
|
- codespell
|
||||||
- checkov@3.2.48
|
- checkov@3.2.139
|
||||||
- git-diff-check
|
- git-diff-check
|
||||||
- isort@5.13.2
|
- isort@5.13.2
|
||||||
- pylint
|
- pylint
|
||||||
- markdownlint@0.39.0
|
- markdownlint@0.41.0
|
||||||
- prettier@3.2.5
|
- prettier@3.3.2
|
||||||
- ruff@0.3.4
|
- ruff@0.4.9
|
||||||
- trivy@0.50.1
|
- trivy@0.52.2
|
||||||
- trufflehog@3.71.0
|
- trufflehog@3.78.1
|
||||||
- yamllint@1.35.1
|
- yamllint@1.35.1
|
||||||
=======
|
|
||||||
>>>>>>> c7dc465 (Parametryzacja "muzyki mojego ludu")
|
|
||||||
actions:
|
actions:
|
||||||
disabled:
|
disabled:
|
||||||
- trunk-announce
|
- trunk-announce
|
||||||
|
|||||||
@@ -0,0 +1,46 @@
|
|||||||
|
import discord
|
||||||
|
from discord.ext import commands
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
class Music(commands.Cog):
|
||||||
|
def __init__(self, bot):
|
||||||
|
self.bot = bot # This is so you can access Bot instance in your cog
|
||||||
|
|
||||||
|
# You must have this function for `bot.load_extension` to call
|
||||||
|
def setup(bot):
|
||||||
|
bot.add_cog(Music(bot))
|
||||||
|
|
||||||
|
@commands.hybrid_command(
|
||||||
|
name="przytul", description="Przytul kogoś - daj mention po komendzie :)"
|
||||||
|
)
|
||||||
|
async def przytul(ctx, arg: Optional[discord.Member] = None):
|
||||||
|
"""
|
||||||
|
Generate a text about hugging mentioned user.
|
||||||
|
|
||||||
|
:param ctx: ctx stands for "context" and is a required parameter in Discord.py commands. It
|
||||||
|
represents the context in which the command was invoked, including information such as the message,
|
||||||
|
the channel, the server, and the user who invoked the command
|
||||||
|
:param arg: arg is a parameter of the function "przytul" that expects a Discord member object. The
|
||||||
|
parameter is optional, meaning that if no member object is provided, it will default to None
|
||||||
|
:type arg: Optional[discord.Member]
|
||||||
|
"""
|
||||||
|
async with ctx.typing():
|
||||||
|
nieprzytulac = False
|
||||||
|
for mention in ctx.message.mentions:
|
||||||
|
for role in mention.roles:
|
||||||
|
if role.name == "NIEPRZYTULAĆ!":
|
||||||
|
nieprzytulac = True
|
||||||
|
|
||||||
|
if arg and nieprzytulac:
|
||||||
|
await ctx.send(
|
||||||
|
f"Żebym ja Ciebie nie przytulił {ctx.message.author.mention}"
|
||||||
|
)
|
||||||
|
elif arg:
|
||||||
|
await ctx.send(
|
||||||
|
# trunk-ignore(codespell/misspelled)
|
||||||
|
f"Już dobrze.... Już dobrze... Ojej.. Biedactwo... :( *W ułamku sekundy {arg.mention} znajduje sie w duszącym uścisku. Żebra trzeszczą - kilka pęka. Pacnięcia po plecach grożą odbiciem nerek, a głaskanie po głowie powoduje wstrząs mózgu*"
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
await ctx.send(
|
||||||
|
"Kogo mam przytulić? *Wyłamuje kostki i przeciąga się - jego 200 kilowa sylwetka złożona z samych mięśni świadczy o tym że jest gotowy*"
|
||||||
|
)
|
||||||
@@ -0,0 +1,173 @@
|
|||||||
|
import logging
|
||||||
|
import threading
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
from queue import Queue, Empty
|
||||||
|
from flask import Flask, jsonify, request
|
||||||
|
from waitress import serve
|
||||||
|
|
||||||
|
HOST_ADDRESS = "192.168.1.191"
|
||||||
|
PORT_ADDRESS = 5000
|
||||||
|
OUT_COMM_Q = Queue()
|
||||||
|
IN_COMM_Q = Queue()
|
||||||
|
awaiting_q = []
|
||||||
|
incoming_q = Queue()
|
||||||
|
app = Flask(__name__)
|
||||||
|
prepped_tracks = {
|
||||||
|
"requests" : "",
|
||||||
|
"hits" : "",
|
||||||
|
"all": "",
|
||||||
|
"prio":"",
|
||||||
|
"jingles":""
|
||||||
|
}
|
||||||
|
|
||||||
|
class QueryControl:
|
||||||
|
"""
|
||||||
|
This class `QueryControl` is used to manage queries with information about the author, UUID,
|
||||||
|
content, logger, context, and replies.
|
||||||
|
"""
|
||||||
|
def __init__(self, query_author, query_uuid, query_content, ctx) -> None:
|
||||||
|
self.author = query_author
|
||||||
|
self.uuid = query_uuid
|
||||||
|
self.content = query_content
|
||||||
|
self.logger = logging.getLogger("discord")
|
||||||
|
self.stop = False
|
||||||
|
self.ctx = ctx
|
||||||
|
self.logger.info(
|
||||||
|
f"Created Query control for {self.author}, {self.uuid}: {self.content}"
|
||||||
|
)
|
||||||
|
self.replies = []
|
||||||
|
|
||||||
|
@app.route("/prepped_tracks", methods=["POST"])
|
||||||
|
def log_radio_tracks():
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info(request)
|
||||||
|
record = json.loads(request.data)
|
||||||
|
logger.info(record)
|
||||||
|
logger.info("DATA RECEIVED")
|
||||||
|
return jsonify("SUCCESS")
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/conjurer", methods=["POST"])
|
||||||
|
def answer_external_command():
|
||||||
|
"""
|
||||||
|
The function `answer_external_command` logs the request data, loads the data as JSON, logs the
|
||||||
|
record, and then puts the record into an incoming queue before returning a success message.
|
||||||
|
:return: The function `answer_external_command()` is returning a JSON response with the message
|
||||||
|
"SUCCESS".
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
logger.info(request)
|
||||||
|
record = json.loads(request.data)
|
||||||
|
logger.info(record)
|
||||||
|
logger.info("DATA RECEIVED")
|
||||||
|
incoming_q.put(record)
|
||||||
|
return jsonify("SUCCESS")
|
||||||
|
|
||||||
|
@app.route("/conjurer", methods=["GET"])
|
||||||
|
def check_alive():
|
||||||
|
"""
|
||||||
|
The function "check_alive" returns a JSON response with the message "ALIVE" when the "/conjurer"
|
||||||
|
route is accessed via a GET request.
|
||||||
|
:return: The code snippet is a Flask route that listens for GET requests to the "/conjurer"
|
||||||
|
endpoint. When a GET request is made to this endpoint, the function check_alive() is called, which
|
||||||
|
returns a JSON response with the message "ALIVE".
|
||||||
|
"""
|
||||||
|
return jsonify("ALIVE")
|
||||||
|
|
||||||
|
def flask_debug():
|
||||||
|
"""
|
||||||
|
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
||||||
|
Do not use for production for fucks sake.
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info("Attempt debug")
|
||||||
|
# trunk-ignore(bandit/B201)
|
||||||
|
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
|
def waitress_run():
|
||||||
|
"""
|
||||||
|
The `waitress_run` function serves the `app` on host "0.0.0.0"
|
||||||
|
and port 5000 using the Waitress WSGI server.
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
|
||||||
|
logger.info("Attempt waitress")
|
||||||
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
def scan_queue():
|
||||||
|
"""
|
||||||
|
The function `scan_queue` reads data from a queue, logs it, and appends it to another queue.
|
||||||
|
|
||||||
|
:param _logger: The `_logger` parameter is typically an instance of a logging object that is used to
|
||||||
|
record and store log messages. It is commonly used to track the flow of the program, record errors,
|
||||||
|
and provide information for debugging purposes. In this code snippet, the `_logger` object is used
|
||||||
|
to log the
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
while True:
|
||||||
|
data = OUT_COMM_Q.get()
|
||||||
|
logger.info(data)
|
||||||
|
awaiting_q.append(data)
|
||||||
|
|
||||||
|
def scan_incoming():
|
||||||
|
"""
|
||||||
|
The `scan_incoming` function continuously checks for incoming data, processes it, and logs when data
|
||||||
|
is found.
|
||||||
|
|
||||||
|
:param _logger: The `_logger` parameter in the `scan_incoming` function is a logger object that is
|
||||||
|
used to log messages or information during the execution of the function. It is typically used for
|
||||||
|
debugging, monitoring, or tracking the flow of the program. In this case, the `_logger` is being
|
||||||
|
used
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
answer = incoming_q.get(block=False)
|
||||||
|
logger.info("DATA FOUND")
|
||||||
|
record_stored = False
|
||||||
|
for record in awaiting_q:
|
||||||
|
if record.uuid in answer.keys():
|
||||||
|
record_stored = True
|
||||||
|
record.stop = True
|
||||||
|
record.entries = answer[record.uuid]
|
||||||
|
IN_COMM_Q.put(record)
|
||||||
|
if not record_stored:
|
||||||
|
for key in answer.keys():
|
||||||
|
record = QueryControl("Orphaned", key, "Orphan", None)
|
||||||
|
record.stop = True
|
||||||
|
record.entries = answer[record.uuid]
|
||||||
|
IN_COMM_Q.put(record)
|
||||||
|
except Empty:
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
|
def comm_subroutine(logger):
|
||||||
|
"""
|
||||||
|
The `comm_subroutine` function starts multiple threads to run different tasks concurrently.
|
||||||
|
|
||||||
|
:param logger: The `logger` parameter in the `comm_subroutine` function is an instance of a logger
|
||||||
|
object that is used for logging messages at various levels (e.g., debug, info, warning, error). In
|
||||||
|
the provided code snippet, the logger is used to log messages at the "info" level
|
||||||
|
"""
|
||||||
|
#logger.setLevel(logging.DEBUG)
|
||||||
|
logger = logging.getLogger("discord")
|
||||||
|
logger.info("Started comms")
|
||||||
|
threads = []
|
||||||
|
#threads.append(threading.Thread(target=flask_debug))
|
||||||
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
|
threads.append(threading.Thread(target=scan_queue))
|
||||||
|
threads.append(threading.Thread(target=scan_incoming))
|
||||||
|
|
||||||
|
|
||||||
|
for worker in threads:
|
||||||
|
worker.start()
|
||||||
|
for worker in threads:
|
||||||
|
worker.join()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
logging_client = logging.getLogger(__name__)
|
||||||
|
comm_subroutine(logging_client)
|
||||||
@@ -1,190 +1,507 @@
|
|||||||
import re
|
"""
|
||||||
|
This module contains the implementation of the Librarian class and
|
||||||
|
related functions for searching and refining queries.
|
||||||
|
|
||||||
|
Classes:
|
||||||
|
- Librarian: Represents a librarian object that performs search and
|
||||||
|
refinement operations on queries
|
||||||
|
|
||||||
|
Functions:
|
||||||
|
- flask_debug: Starts a Flask application in debug mode without using the reloader.
|
||||||
|
- waitress_run: Serves the Flask application using the Waitress WSGI server.
|
||||||
|
- BackgroundTaskSearch: Represents a background task for running the
|
||||||
|
Librarian object asynchronously
|
||||||
|
"""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import json
|
||||||
import logging
|
import logging
|
||||||
|
import netrc
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
from flask import Flask, jsonify, request
|
|
||||||
import netrc
|
|
||||||
import json
|
|
||||||
from json.decoder import JSONDecodeError
|
from json.decoder import JSONDecodeError
|
||||||
from platform import uname
|
from logging import handlers
|
||||||
from sys import platform
|
from queue import Queue
|
||||||
from urllib.request import urlopen
|
|
||||||
|
import requests
|
||||||
|
import scrape_bot
|
||||||
|
import search_bot
|
||||||
|
#import search_bot2 as search_bot
|
||||||
|
from flask import Flask, jsonify, request
|
||||||
from habanero import Crossref
|
from habanero import Crossref
|
||||||
from waitress import serve
|
from waitress import serve
|
||||||
|
|
||||||
if platform in ("linux", "linux2"):
|
# Constants
|
||||||
SEPARATOR_FILE_PATH = "/"
|
NETRC_FILE = r"C:\Users\Activcom.pl\.netrc"
|
||||||
if "microsoft-standard" in uname().release:
|
HOST_ADDRESS = "192.168.1.192"
|
||||||
NETRC_FILE = "/home/mtuszowski/.netrc"
|
|
||||||
else:
|
|
||||||
NETRC_FILE = "/home/pi/.netrc"
|
|
||||||
elif platform == "win32":
|
|
||||||
NETRC_FILE = "C:\\Users\\mtusz\\.netrc"
|
|
||||||
|
|
||||||
#HOST_ADDRESS = "192.168.1.191"
|
|
||||||
HOST_ADDRESS = "0.0.0.0"
|
|
||||||
PORT_ADDRESS = 5001
|
PORT_ADDRESS = 5001
|
||||||
|
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
||||||
|
SEND_RESULTS = "/conjurer"
|
||||||
|
BDSM_UUID_TEST = "96b7f85a-1142-4908-8986-62a2ea25a147"
|
||||||
|
|
||||||
|
MAX_CR_RESULTS = 1000
|
||||||
|
#TEST PURPOSES ONLY!
|
||||||
|
#MAX_CR_RESULTS = 5
|
||||||
|
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
|
||||||
|
app = Flask(__name__)
|
||||||
|
|
||||||
|
librarian_queue = Queue()
|
||||||
|
librarian_list = []
|
||||||
|
|
||||||
|
|
||||||
|
# trunk-ignore(pylint/R0902)
|
||||||
class Librarian(object):
|
class Librarian(object):
|
||||||
|
"""
|
||||||
|
Represents a librarian object that performs search and refinement operations on queries.
|
||||||
|
"""
|
||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self, _app, query, uuid, _deep_search) -> None:
|
||||||
|
"""
|
||||||
|
Initializes a Librarian object.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
- _app: The Flask application object.
|
||||||
|
- query: The query to be searched.
|
||||||
|
- uuid: The unique identifier for the search.
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
- cr: The Crossref object for performing the search.
|
||||||
|
- query: The query to be searched.
|
||||||
|
- uuid: The unique identifier for the search.
|
||||||
|
- limit: The maximum number of search results to fetch.
|
||||||
|
- fetched: The number of search results fetched so far.
|
||||||
|
- hit: The number of search results that match the refinement criteria.
|
||||||
|
- total: The total number of search results.
|
||||||
|
- app: The Flask application object.
|
||||||
|
- live_results: A list to store live search results.
|
||||||
|
- final_result: A list to store the final refined search results.
|
||||||
|
- not_in_db: A list to store search results that are not in the local database.
|
||||||
|
- search_result_from_cr: A dictionary to store the search results from Crossref.
|
||||||
|
- done: A flag indicating if the search is done.
|
||||||
|
"""
|
||||||
netrc_mod = netrc.netrc(NETRC_FILE)
|
netrc_mod = netrc.netrc(NETRC_FILE)
|
||||||
authTokens = netrc_mod.authenticators("crossref")
|
auth_tokens = netrc_mod.authenticators("crossref")
|
||||||
self.cr = Crossref(mailto=authTokens[0])
|
self.cr = Crossref(
|
||||||
self.limit = 1000
|
mailto=auth_tokens[0],
|
||||||
|
ua_string=f"Conjurer project. mailto:{auth_tokens[0]}"
|
||||||
|
)
|
||||||
|
self.query = query
|
||||||
|
self.uuid = str(uuid)
|
||||||
|
self.limit = MAX_CR_RESULTS
|
||||||
self.fetched = 0
|
self.fetched = 0
|
||||||
self.hit = 0
|
self.hit = 0
|
||||||
self.total = 0
|
self.total = 0
|
||||||
|
self.app = _app
|
||||||
|
self.live_results = []
|
||||||
|
self.final_result = {}
|
||||||
|
self.not_in_db = {}
|
||||||
|
self.search_result_from_cr = {}
|
||||||
|
self.done = False
|
||||||
|
self.deep_search = _deep_search
|
||||||
|
|
||||||
def search_crossref(self, query, uuid):
|
async def search_crossref(self, query, deep_search=False):
|
||||||
search_result_from_cr = self.cr.works(query = query, limit = self.limit)
|
"""
|
||||||
self.total = search_result_from_cr['message']['total-results']
|
Performs a search on Crossref for the given query.
|
||||||
self.fetched = len(search_result_from_cr['message']['items'])
|
|
||||||
#add to sql with UUID
|
Args:
|
||||||
#sql row - title, doi, author, uuid, last retrieved, is_available
|
- query: The query to be searched.
|
||||||
with open("results.json", "r+", encoding="UTF-8") as data_file:
|
|
||||||
|
Returns:
|
||||||
|
- result: The search result from Crossref.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
- None.
|
||||||
|
"""
|
||||||
|
|
||||||
|
self.app.logger.info("STARTED SEARCH")
|
||||||
|
|
||||||
|
if not deep_search:
|
||||||
|
query_limit = MAX_CR_RESULTS if MAX_CR_RESULTS < 1000 else 1000
|
||||||
|
cr_result = self.cr.works(query=query, limit=query_limit)
|
||||||
|
self.search_result_from_cr.update(cr_result)
|
||||||
|
self.total = cr_result["message"]["total-results"]
|
||||||
|
self.fetched += len(cr_result["message"]["items"])
|
||||||
|
self.app.logger.info(self.total)
|
||||||
|
self.app.logger.info(self.fetched)
|
||||||
|
while self.total > self.fetched and self.limit > self.fetched:
|
||||||
|
tmp_result = self.cr.works(query=query, limit=query_limit, offset=self.fetched)
|
||||||
|
cr_result["message"]["items"].extend(tmp_result["message"]["items"])
|
||||||
|
self.total = tmp_result["message"]["total-results"]
|
||||||
|
self.fetched = len(cr_result["message"]["items"])
|
||||||
|
self.app.logger.info(self.total)
|
||||||
|
self.app.logger.info(self.fetched)
|
||||||
|
time.sleep(0.1)
|
||||||
|
|
||||||
|
else:
|
||||||
|
cr_result = self.cr.works(query=query, cursor_max=15000, cursor='*', progress_bar = True)
|
||||||
|
result = cr_result[0]
|
||||||
|
for item in cr_result[1:]:
|
||||||
|
result["message"]["items"].extend(item["message"]["items"])
|
||||||
|
self.total = item["message"]["total-results"]
|
||||||
|
self.fetched = len(result["message"]["items"])
|
||||||
|
self.app.logger.info(self.total)
|
||||||
|
self.app.logger.info(self.fetched)
|
||||||
|
cr_result = result
|
||||||
|
self.app.logger.info("Total, fetched:")
|
||||||
|
self.app.logger.info(self.total)
|
||||||
|
self.app.logger.info(self.fetched)
|
||||||
|
self.search_result_from_cr.update(cr_result)
|
||||||
|
self.total = cr_result["message"]["total-results"]
|
||||||
|
self.app.logger.info("CROSSREF DONE")
|
||||||
|
|
||||||
|
self.app.logger.info("CROSSREF DONE")
|
||||||
|
with open("cr_results.json", "r+", encoding="utf-8") as data_file:
|
||||||
# First we load existing data into a dict.
|
# First we load existing data into a dict.
|
||||||
try:
|
try:
|
||||||
file_data = json.load(data_file)
|
file_data = json.load(data_file)
|
||||||
except JSONDecodeError:
|
except JSONDecodeError:
|
||||||
file_data = []
|
file_data = {}
|
||||||
# Join new_data with file_data inside emp_details
|
data_file.truncate(0)
|
||||||
summarized_results = []
|
|
||||||
for item in search_result_from_cr['message']['items']:
|
|
||||||
container = "NO"
|
|
||||||
if "container-title" in item:
|
|
||||||
container = "YES"
|
|
||||||
summarized_results.append({"DOI" : item['DOI'], "title": item['title'] if 'title' in item else None,"type": item['type'] if 'type' in item else None, "container": container})
|
|
||||||
result = {"UUID": uuid, "total_results": search_result_from_cr['message']['total-results'], "on_page" : 1, "summary":summarized_results, "results":search_result_from_cr['message']['items'], }
|
|
||||||
file_data.append(result)
|
|
||||||
print("Saving")
|
|
||||||
data_file.seek(0)
|
data_file.seek(0)
|
||||||
# convert back to json.
|
tmp = {self.uuid : self.search_result_from_cr}
|
||||||
|
if file_data:
|
||||||
|
file_data.update(tmp)
|
||||||
|
else:
|
||||||
|
file_data = tmp
|
||||||
json.dump(file_data, data_file, indent=4)
|
json.dump(file_data, data_file, indent=4)
|
||||||
return result
|
return cr_result
|
||||||
|
|
||||||
|
|
||||||
def check_if_exists(self,page_url):
|
async def refine_search(self, unrefined_result):
|
||||||
#TODO: Spowolnic
|
"""
|
||||||
#TODO: Sprawdzac po mirrorach
|
Refines the search query based on the unrefined search result.
|
||||||
#TODO: sprawdzac w tym jebanym pliku
|
|
||||||
in_scihub_db = True
|
|
||||||
if not page_url.startswith(("http:", "https:")):
|
|
||||||
raise ValueError("URL must start with 'http:' or 'https:'")
|
|
||||||
# trunk-ignore(bandit/B310)
|
|
||||||
with urlopen(page_url) as response:
|
|
||||||
data = response.read()
|
|
||||||
text = data.decode("utf-8")
|
|
||||||
for line in text.splitlines():
|
|
||||||
if re.match(r".*Unfortunately, Sci-Hub doesn't have the requested document.*", line):
|
|
||||||
return None
|
|
||||||
if in_scihub_db:
|
|
||||||
return text
|
|
||||||
return None
|
|
||||||
|
|
||||||
def download(self, page_url):
|
Args:
|
||||||
if response := self.check_if_exists(page_url):
|
- unrefined_result: The unrefined search result.
|
||||||
for line in response.splitlines():
|
|
||||||
if m := re.match(r".*<embed type=\"application.pdf\"\s*src=\"(.*\.pdf)",line):
|
|
||||||
return "https:" + str(m.group(1))
|
|
||||||
return None
|
|
||||||
|
|
||||||
def provide_sci_hub_link(self, page_url):
|
Returns:
|
||||||
if self.check_if_exists(page_url):
|
- refined_result: The refined search result.
|
||||||
return page_url
|
|
||||||
return None
|
|
||||||
|
|
||||||
def refine_query(self, unrefined_reqult):
|
Raises:
|
||||||
print("Refine result")
|
- None.
|
||||||
refined_result = []
|
"""
|
||||||
for item in unrefined_reqult["summary"]:
|
summarized_results = []
|
||||||
if item["container"] == "NO" and item["title"]:
|
self.app.logger.info("REFINE: Removing all derived works from the list")
|
||||||
item_link = "https://sci-hub.se/" + str(item['DOI'])
|
for item in unrefined_result["message"]["items"]:
|
||||||
if self.check_if_exists(item_link):
|
summarized_results.append(
|
||||||
refined_result.append(item)
|
{
|
||||||
print ("HIT")
|
"DOI": item["DOI"],
|
||||||
print(item["title"])
|
"title": item["title"] if "title" in item else None,
|
||||||
time.sleep(0.01)
|
"type": item["type"] if "type" in item else None,
|
||||||
print("Existing hits")
|
}
|
||||||
self.hit = len(refined_result)
|
)
|
||||||
|
|
||||||
|
partial_result = {
|
||||||
|
self.uuid: {
|
||||||
|
"total_results": self.search_result_from_cr["message"][
|
||||||
|
"total-results"
|
||||||
|
],
|
||||||
|
"on_page": 1,
|
||||||
|
"summary": summarized_results,
|
||||||
|
"results": self.search_result_from_cr["message"]["items"],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
self.app.logger.info("REFINE: Dumping to file")
|
||||||
|
temp = []
|
||||||
|
refined_result = {}
|
||||||
|
for key in partial_result:
|
||||||
|
self.app.logger.info("KEY:")
|
||||||
|
self.app.logger.info(key)
|
||||||
|
|
||||||
|
for item in partial_result[self.uuid]["summary"]:
|
||||||
|
if item["title"]:
|
||||||
|
temp.append(item)
|
||||||
|
|
||||||
|
for item in temp:
|
||||||
|
refined_result[item["DOI"]]= item
|
||||||
|
with open("rr_results.json", "r+", encoding="utf-8") as data_file:
|
||||||
|
# First we load existing data into a dict.
|
||||||
|
try:
|
||||||
|
file_data = json.load(data_file)
|
||||||
|
except JSONDecodeError:
|
||||||
|
file_data = {}
|
||||||
|
data_file.truncate(0)
|
||||||
|
data_file.seek(0)
|
||||||
|
tmp = {self.uuid: refined_result}
|
||||||
|
if file_data:
|
||||||
|
file_data.update(tmp)
|
||||||
|
else:
|
||||||
|
file_data = tmp
|
||||||
|
json.dump(file_data, data_file, indent=4)
|
||||||
return refined_result
|
return refined_result
|
||||||
|
|
||||||
def answer_query(self, query, uuid):
|
async def check_if_exists(self, refined_result):
|
||||||
#Search sql for UUID
|
"""
|
||||||
#SQL SELECT * FROM RESULTS WHERE UUID = uuid
|
Checks if the given DOI exists.
|
||||||
print("Answer result")
|
|
||||||
uuid_found = False
|
|
||||||
database = None
|
|
||||||
with open("results.json", "r", encoding="UTF-8") as new_file:
|
|
||||||
#tu logika komunikacji z SQL i cała logika związana z wyszukaniem
|
|
||||||
#ale tymczasowo plik
|
|
||||||
try:
|
|
||||||
database = json.load(new_file)
|
|
||||||
except JSONDecodeError:
|
|
||||||
pass
|
|
||||||
if database:
|
|
||||||
for item in database:
|
|
||||||
if item["UUID"] == uuid:
|
|
||||||
print (item["UUID"])
|
|
||||||
uuid_found = True
|
|
||||||
return self.refine_query(item)
|
|
||||||
|
|
||||||
if not uuid_found:
|
Args:
|
||||||
answer = self.search_crossref(query=query, uuid=uuid)
|
- doi: The DOI to be checked.
|
||||||
return self.refine_query(answer)
|
- brute_force: A flag indicating if brute force method should be used.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
- result: The search result.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
- None.
|
||||||
|
"""
|
||||||
|
result = {}
|
||||||
|
self.app.logger.info("REFINE: Running search in the backend app")
|
||||||
|
dois = []
|
||||||
|
for item, value in refined_result.items():
|
||||||
|
dois.append([item, value])
|
||||||
|
coro = asyncio.to_thread(
|
||||||
|
search_bot.search_for_doi, dois, self.live_results, self.app.logger
|
||||||
|
)
|
||||||
|
result = await coro
|
||||||
|
result_list = []
|
||||||
|
result_no_db = []
|
||||||
|
for item in result:
|
||||||
|
if item["exists"]:
|
||||||
|
result_list.append(item)
|
||||||
|
else:
|
||||||
|
result_no_db.append(item)
|
||||||
|
self.hit = len(result)
|
||||||
|
return result_list, result_no_db
|
||||||
|
|
||||||
|
|
||||||
|
async def answer_query(self, deep_search=False):
|
||||||
|
"""
|
||||||
|
Answers the search query.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
- None.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
- result: The search result.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
- None.
|
||||||
|
"""
|
||||||
|
self.app.logger.info(f"Search started {self.uuid}")
|
||||||
|
cr_result = await self.search_crossref(query=self.query, deep_search=deep_search)
|
||||||
|
refined_result = await self.refine_search(cr_result)
|
||||||
|
answer, negative_answer = await self.check_if_exists(refined_result)
|
||||||
|
|
||||||
|
self.app.logger.info("Returning result")
|
||||||
|
self.app.logger.info(answer)
|
||||||
|
self.app.logger.info(negative_answer)
|
||||||
|
|
||||||
|
for item in answer:
|
||||||
|
self.final_result[item["DOI"]] = {"Title": item["data"]["title"], "type": item["data"]["type"]}
|
||||||
|
for item in negative_answer:
|
||||||
|
self.not_in_db[item["DOI"]] = {"Title": item["data"]["title"], "type": item["data"]["type"]}
|
||||||
|
self.app.logger.info("Returning result case2")
|
||||||
|
self.app.logger.info(self.final_result)
|
||||||
|
return self.final_result
|
||||||
|
|
||||||
|
# ============================= FLASK INTERNALS===============================
|
||||||
|
|
||||||
#example not exists
|
|
||||||
#sample_non_existing_page_url = "https://sci-hub.se/10.4324/9781003006237"
|
|
||||||
#sample usage to download
|
|
||||||
#sample_existing_page_url = "https://sci-hub.se/10.1057/9781137435026"
|
|
||||||
#print(cl.download(sample_existing_page_url))
|
|
||||||
#cl.search_crossref("BDSM")
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
|
||||||
def flask_debug():
|
def flask_debug():
|
||||||
"""
|
"""
|
||||||
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
Starts a Flask application in debug mode without using the reloader.
|
||||||
Do not use for production for fucks sake.
|
|
||||||
|
Args:
|
||||||
|
- None.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
- None.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
- None.
|
||||||
"""
|
"""
|
||||||
# trunk-ignore(bandit/B201)
|
# trunk-ignore(bandit/B201)
|
||||||
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
def waitress_run():
|
def waitress_run():
|
||||||
"""
|
"""
|
||||||
The `waitress_run` function serves the `app` on host "0.0.0.0"
|
Serves the Flask application using the Waitress WSGI server.
|
||||||
and port 5000 using the Waitress WSGI server.
|
|
||||||
|
Args:
|
||||||
|
- None.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
- None.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
- None.
|
||||||
"""
|
"""
|
||||||
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
|
class BackgroundTaskSearch(threading.Thread):
|
||||||
|
"""
|
||||||
|
A background task for searching and saving results to files.
|
||||||
|
|
||||||
|
This class extends the `threading.Thread` class and is responsible for running
|
||||||
|
the search task in the background. It retrieves queries from a queue, performs
|
||||||
|
the search, and saves the results to files.
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
app (App): The application instance.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
"""
|
||||||
|
Run the background task.
|
||||||
|
|
||||||
|
This method is called when the thread is started. It creates a new event loop,
|
||||||
|
runs the `_run` method, and closes the event loop.
|
||||||
|
"""
|
||||||
|
loop = asyncio.new_event_loop()
|
||||||
|
loop.run_until_complete(self._run())
|
||||||
|
loop.close()
|
||||||
|
|
||||||
|
async def _run(self):
|
||||||
|
"""
|
||||||
|
Perform the search task.
|
||||||
|
|
||||||
|
This method is an asynchronous coroutine that runs in a loop. It retrieves a
|
||||||
|
librarian from the queue, answers the query, and saves the results to files.
|
||||||
|
It also sends the results to a remote server.
|
||||||
|
|
||||||
|
The search task continues running indefinitely until the thread is stopped.
|
||||||
|
"""
|
||||||
|
while True:
|
||||||
|
database = None
|
||||||
|
ndb_database = None
|
||||||
|
librarian = librarian_queue.get()
|
||||||
|
self.app.logger.info("STARTED")
|
||||||
|
result = await librarian.answer_query(librarian.deep_search)
|
||||||
|
result = {librarian.uuid: result}
|
||||||
|
self.app.logger.info("Saving to file")
|
||||||
|
|
||||||
|
# Save results to "not_in_db.json" file
|
||||||
|
with open("not_in_db.json", "r+", encoding="utf-8") as ndb_file:
|
||||||
|
try:
|
||||||
|
ndb_database = json.load(ndb_file)
|
||||||
|
except JSONDecodeError:
|
||||||
|
pass
|
||||||
|
if ndb_database:
|
||||||
|
ndb_database.update(librarian.not_in_db)
|
||||||
|
else:
|
||||||
|
ndb_database = librarian.not_in_db
|
||||||
|
ndb_file.truncate(0)
|
||||||
|
ndb_file.seek(0)
|
||||||
|
json.dump(ndb_database, ndb_file)
|
||||||
|
|
||||||
|
# Save results to "s_results.json" file
|
||||||
|
with open("s_results.json", "r+", encoding="utf-8") as s_file:
|
||||||
|
try:
|
||||||
|
database = json.load(s_file)
|
||||||
|
except JSONDecodeError:
|
||||||
|
pass
|
||||||
|
if database:
|
||||||
|
self.app.logger.info(database)
|
||||||
|
self.app.logger.info(result)
|
||||||
|
database.update(result)
|
||||||
|
else:
|
||||||
|
database = result
|
||||||
|
self.app.logger.info("DUMPING DATA")
|
||||||
|
s_file.truncate(0)
|
||||||
|
s_file.seek(0)
|
||||||
|
json.dump(database, s_file)
|
||||||
|
self.app.logger.info("FINISHED")
|
||||||
|
|
||||||
|
self.app.logger.info(result)
|
||||||
|
coroutine = asyncio.to_thread(
|
||||||
|
requests.post,
|
||||||
|
f"{MAIN_BOT_ADDRESS}{SEND_RESULTS}",
|
||||||
|
json=result,
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
self.app.logger.info("SENT")
|
||||||
|
result = await coroutine
|
||||||
|
self.app.logger.info(result.status_code)
|
||||||
|
self.app.logger.info("SEND CONFIRMED")
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
|
# ==================================SERVER ROUTES==========================================
|
||||||
@app.route("/query", methods=["POST"])
|
@app.route("/query", methods=["POST"])
|
||||||
def query_database():
|
async def query_database():
|
||||||
|
"""
|
||||||
|
Endpoint for querying the database.
|
||||||
|
|
||||||
|
This function receives a POST request containing a JSON payload with a query and a UUID.
|
||||||
|
It creates a Librarian object with the query and UUID,
|
||||||
|
and adds it to the librarian_queue and librarian_list.
|
||||||
|
Finally, it returns a JSON response indicating the success
|
||||||
|
of the operation, along with the query, UUID,
|
||||||
|
and the current size of the librarian_queue.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
tuple: A tuple containing a JSON response and a status code.
|
||||||
|
"""
|
||||||
record = json.loads(request.data)
|
record = json.loads(request.data)
|
||||||
app.logger.info(record)
|
app.logger.info(record)
|
||||||
app.logger.info(record["query"])
|
app.logger.info(record["query"])
|
||||||
app.logger.info(record["UUID"])
|
app.logger.info(record["UUID"])
|
||||||
cl = Librarian()
|
uuid = record["UUID"]
|
||||||
answer_data = cl.answer_query(record["query"], record["UUID"])
|
deep_search = record["deep_search"]
|
||||||
app.logger.info(answer_data)
|
cl = Librarian(app, record["query"], uuid, deep_search)
|
||||||
summary = (cl.hit,cl.fetched, cl.total, record["UUID"])
|
librarian_queue.put(cl)
|
||||||
answer_data = (summary,answer_data)
|
librarian_list.append(cl)
|
||||||
|
answer_data = (record["query"], record["UUID"], librarian_queue.qsize())
|
||||||
return_data = (
|
return_data = (
|
||||||
jsonify(isError=False, message="Success", statusCode=200, data=answer_data),
|
jsonify(isError=False, message="Success", statusCode=200, data=answer_data),
|
||||||
200,
|
200,
|
||||||
)
|
)
|
||||||
return return_data
|
return return_data
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
logger = logging.getLogger('conjurer_musician')
|
|
||||||
logger.setLevel(logging.DEBUG)
|
|
||||||
threads = []
|
|
||||||
#threads.append(threading.Thread(target=waitress_run))
|
|
||||||
threads.append(threading.Thread(target=flask_debug))
|
|
||||||
|
|
||||||
|
@app.route("/get_partial_result", methods=["POST"])
|
||||||
|
async def get_partial():
|
||||||
|
"""
|
||||||
|
Retrieves the partial result for a given UUID.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
A JSON response containing the partial result.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
for lib in librarian_list:
|
||||||
|
if lib.uuid == record["UUID"]:
|
||||||
|
answer_data = lib.live_results
|
||||||
|
break
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data=answer_data),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
|
# =======================================MAIN===================================================
|
||||||
|
if __name__ == "__main__":
|
||||||
|
app.logger.setLevel(logging.DEBUG)
|
||||||
|
h1 = handlers.RotatingFileHandler(
|
||||||
|
filename="D:\\logs\\librarian.log",
|
||||||
|
encoding="utf-8",
|
||||||
|
mode="a",
|
||||||
|
maxBytes=6 * 1024 * 1024,
|
||||||
|
backupCount=6,
|
||||||
|
)
|
||||||
|
|
||||||
|
app.logger.addHandler(h1)
|
||||||
|
threads = []
|
||||||
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
|
# threads.append(threading.Thread(target=flask_debug))
|
||||||
|
bgtask = BackgroundTaskSearch()
|
||||||
|
bgtask.app = app
|
||||||
|
threads.append(bgtask)
|
||||||
|
threads.append(threading.Thread(target=scrape_bot.scraper, args=(app.logger,)))
|
||||||
|
i = 0
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.start()
|
try:
|
||||||
|
app.logger.info("App number: %s", i)
|
||||||
|
i += 1
|
||||||
|
worker.start()
|
||||||
|
except RuntimeError as e:
|
||||||
|
app.logger.error("Exploded")
|
||||||
|
print(str(e))
|
||||||
for worker in threads:
|
for worker in threads:
|
||||||
worker.join()
|
worker.join()
|
||||||
|
|
||||||
#1. robi liste wyników i ją wyświetla - ale tylko tytuły niezależnych dzieł - jeśli jest coś w containerze to trza pogrupować
|
|
||||||
#2. Tworzy przyciski pozwalające sprawdzić czy jest dostępne - i jeśli jest zwraca link
|
|
||||||
#3. Jest też przycisk "more" - wyszukujemy po 100, wyświetlamy po 20. Po 100 jak jeszcze to dopiero odpytujemy crossref przyciski strzałek przesuwają po obecnej porcji danych
|
|
||||||
|
|||||||
@@ -0,0 +1,8 @@
|
|||||||
|
|
||||||
|
FILEPTH = r"C:\Database\chunks\4_chunk.txt"
|
||||||
|
|
||||||
|
with open(FILEPTH, "r", encoding="utf-8") as file:
|
||||||
|
tab = file.readlines()
|
||||||
|
print(len(tab))
|
||||||
|
|
||||||
|
#1842295
|
||||||
@@ -1,11 +1,33 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
python3 -m venv /home/pi/Conjurer_librarian/env
|
# This script sets up a virtual environment for the Conjurer Librarian service and installs the required dependencies.
|
||||||
cp /home/pi/conjurer/conjurer_librarian/conjurer_librarian.py ./Conjurer_librarian/
|
|
||||||
cp /home/pi/conjurer/conjurer_librarian/requirements_librarian.txt ./Conjurer_librarian/
|
|
||||||
cd /home/pi/Conjurer_librarian
|
|
||||||
source /home/pi/Conjurer_librarian/env/bin/activate
|
|
||||||
./env/bin/python3 -m pip install --upgrade pip
|
|
||||||
./env/bin/python3 -m pip install -r requirements_librarian.txt
|
|
||||||
deactivate
|
|
||||||
|
|
||||||
cp /home/pi/conjurer/conjurer_librarian/conjurer_librarian.py ./Conjurer_librarian/
|
|
||||||
|
# Create a virtual environment at /home/pi/Conjurer_librarian/env
|
||||||
|
python3 -m venv /home/pi/Conjurer_librarian/env
|
||||||
|
|
||||||
|
# Copy the Conjurer Librarian Python script to the appropriate directory
|
||||||
|
cp /home/pi/conjurer/conjurer_librarian/conjurer_librarian.py ./Conjurer_librarian/
|
||||||
|
|
||||||
|
# Copy the requirements file to the appropriate directory
|
||||||
|
cp /home/pi/conjurer/conjurer_librarian/requirements_librarian.txt ./Conjurer_librarian/
|
||||||
|
|
||||||
|
# Change directory to /home/pi/Conjurer_librarian
|
||||||
|
cd /home/pi/Conjurer_librarian
|
||||||
|
|
||||||
|
touch /home/pi/Conjurer/rr_results.json
|
||||||
|
touch /home/pi/Conjurer/cr_results.json
|
||||||
|
touch /home/pi/Conjurer/s_results.json
|
||||||
|
touch /home/pi/Conjurer/rr_results.json
|
||||||
|
touch /home/pi/Conjurer/not_in_db.json
|
||||||
|
|
||||||
|
# Activate the virtual environment
|
||||||
|
source /home/pi/Conjurer_librarian/env/bin/activate
|
||||||
|
|
||||||
|
# Upgrade pip
|
||||||
|
./env/bin/python3 -m pip install --upgrade pip
|
||||||
|
|
||||||
|
# Install the required dependencies
|
||||||
|
./env/bin/python3 -m pip install -r requirements_librarian.txt
|
||||||
|
|
||||||
|
# Deactivate the virtual environment
|
||||||
|
deactivate
|
||||||
|
|||||||
@@ -1,3 +1,3 @@
|
|||||||
habanero
|
habanero
|
||||||
waitress
|
waitress
|
||||||
flask
|
flask[async]
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
c:\conjurer\env\Scripts\activate.bat && c:\conjurer\conjurer_librarian\conjurer_librarian.py
|
||||||
@@ -0,0 +1,136 @@
|
|||||||
|
"""
|
||||||
|
This module contains the code for the scrape bot.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import random
|
||||||
|
import re
|
||||||
|
import time
|
||||||
|
from json import JSONDecodeError
|
||||||
|
from queue import Queue
|
||||||
|
from threading import Thread
|
||||||
|
from urllib.request import urlopen
|
||||||
|
|
||||||
|
from requests import ConnectionError as RequestsConnectionError
|
||||||
|
from requests import ConnectTimeout, Timeout
|
||||||
|
|
||||||
|
SCR_DATABASE_PATH = r"C:\\Database\\chunks\\"
|
||||||
|
SCR_FILENAME = "40_chunk.txt"
|
||||||
|
SCR_ENCODING = "utf-8"
|
||||||
|
|
||||||
|
WORK_Q = Queue()
|
||||||
|
random.seed()
|
||||||
|
|
||||||
|
|
||||||
|
def load_ndb_to_q(logger):
|
||||||
|
"""
|
||||||
|
This function loads items from the not_in_db.json file into the work queue.
|
||||||
|
"""
|
||||||
|
logger.info("Loader started")
|
||||||
|
while True:
|
||||||
|
with open("not_in_db.json", "r+", encoding="utf-8") as ndb_file:
|
||||||
|
try:
|
||||||
|
ndb_database = json.load(ndb_file)
|
||||||
|
for _ in range (1,10):
|
||||||
|
try:
|
||||||
|
key = next(iter(ndb_database))
|
||||||
|
_ = ndb_database.pop(key)
|
||||||
|
logger.info(key)
|
||||||
|
url = f"https://sci-hub.se/{key}"
|
||||||
|
WORK_Q.put([key, url, False])
|
||||||
|
except StopIteration:
|
||||||
|
break
|
||||||
|
ndb_file.truncate(0)
|
||||||
|
ndb_file.seek(0)
|
||||||
|
json.dump(ndb_database, ndb_file, indent=4)
|
||||||
|
except JSONDecodeError:
|
||||||
|
time.sleep(60 * 60 * 3)
|
||||||
|
time.sleep(60*60*3)
|
||||||
|
|
||||||
|
|
||||||
|
def check_if_exists_brute_force(logger):
|
||||||
|
"""
|
||||||
|
This function checks if a document exists using brute force search.
|
||||||
|
"""
|
||||||
|
# Function code here
|
||||||
|
while True:
|
||||||
|
logger.info("Scraper tick")
|
||||||
|
item = WORK_Q.get()
|
||||||
|
page_url = item[1]
|
||||||
|
logger.error("REFINE: Brute force search!")
|
||||||
|
if not page_url.startswith(("http:", "https:")):
|
||||||
|
raise ValueError("URL must start with 'http:' or 'https:'")
|
||||||
|
blocked = True
|
||||||
|
try:
|
||||||
|
# trunk-ignore(bandit/B310)
|
||||||
|
with urlopen(page_url) as response:
|
||||||
|
data = response.read()
|
||||||
|
text = data.decode("utf-8")
|
||||||
|
if item[2]:
|
||||||
|
logger.info("Already found")
|
||||||
|
time.sleep(180)
|
||||||
|
blocked = False
|
||||||
|
else:
|
||||||
|
for line in text.splitlines():
|
||||||
|
if re.match(
|
||||||
|
r".*Unfortunately, Sci-Hub doesn't have the requested document.*",
|
||||||
|
line,
|
||||||
|
):
|
||||||
|
blocked = False
|
||||||
|
logger.info("Not found")
|
||||||
|
item[2] = False
|
||||||
|
if m := re.match(
|
||||||
|
r".*<embed type=\"application.pdf\"\s*src=\"(.*\.pdf)", line
|
||||||
|
):
|
||||||
|
blocked = False
|
||||||
|
direct_download_link = "https:" + str(m.group(1))
|
||||||
|
logger.info(direct_download_link)
|
||||||
|
with open(
|
||||||
|
SCR_DATABASE_PATH + SCR_FILENAME,
|
||||||
|
"a",
|
||||||
|
encoding=SCR_ENCODING,
|
||||||
|
) as operated_file:
|
||||||
|
operated_file.write("\n")
|
||||||
|
operated_file.write(item[0])
|
||||||
|
item[2] = True
|
||||||
|
except (
|
||||||
|
Timeout,
|
||||||
|
ConnectTimeout,
|
||||||
|
ConnectionRefusedError,
|
||||||
|
ConnectionError,
|
||||||
|
RequestsConnectionError,
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
if blocked:
|
||||||
|
logger.info(item)
|
||||||
|
logger.info(text)
|
||||||
|
logger.error("Got blocked. Fuck.")
|
||||||
|
time.sleep(60 * 60 * 72)
|
||||||
|
# trunk-ignore(bandit/B311)
|
||||||
|
rand = random.randint(1, 60)
|
||||||
|
logger.info(f"Sleeping for {2*rand} minutes")
|
||||||
|
time.sleep(120 * rand)
|
||||||
|
|
||||||
|
|
||||||
|
def scraper(logger=None):
|
||||||
|
"""
|
||||||
|
This function is responsible for scraping data.
|
||||||
|
"""
|
||||||
|
if not logger:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel("DEBUG")
|
||||||
|
h1 = logging.StreamHandler()
|
||||||
|
logger.addHandler(h1)
|
||||||
|
logger.info("No logger set. Test run")
|
||||||
|
|
||||||
|
loader = Thread(target=load_ndb_to_q, args=(logger,))
|
||||||
|
_scraper = Thread(target=check_if_exists_brute_force, args=(logger,))
|
||||||
|
loader.start()
|
||||||
|
_scraper.start()
|
||||||
|
loader.join()
|
||||||
|
_scraper.join()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
scraper()
|
||||||
@@ -0,0 +1,200 @@
|
|||||||
|
"""
|
||||||
|
This module contains functions for searching for DOI (Digital Object Identifier)
|
||||||
|
in a list of live results.
|
||||||
|
It includes a producer-consumer pattern implementation using threads and queues.
|
||||||
|
|
||||||
|
Functions:
|
||||||
|
- producer: Reads lines from a file and puts them into an output queue.
|
||||||
|
- consumer: Consumes items from an input queue and checks if DOI exists in the live results list.
|
||||||
|
- search_for_doi: Searches for DOI in live results using the producer-consumer pattern.
|
||||||
|
|
||||||
|
Global Variables:
|
||||||
|
- MAXTHREADS: Maximum number of worker threads.
|
||||||
|
- DATABASE_PATH: Path to the database files.
|
||||||
|
- ENCODING: Encoding of the database files.
|
||||||
|
- CHUNK: File name pattern for the database files.
|
||||||
|
- _sentinel: Sentinel object used to signal termination.
|
||||||
|
- result_list: List to store the search results.
|
||||||
|
- WORK_Q_SIZE: Maximum size of the work queue.
|
||||||
|
"""
|
||||||
|
|
||||||
|
# TODO: Wpiemdolić to wszystko w klasę z loggerem przysłanym z góry
|
||||||
|
from queue import Empty, Queue
|
||||||
|
from threading import Thread
|
||||||
|
import time
|
||||||
|
q = Queue()
|
||||||
|
#TODO: Count number of lines in files and print to approximate on which part of the file search is
|
||||||
|
|
||||||
|
# DATA FOR TEST ONLY
|
||||||
|
# MAXTHREADS = 5
|
||||||
|
# DATABASE_PATH = r'C:\\Database\\chunks_1\\'
|
||||||
|
|
||||||
|
# DEPLOYMENT DATA
|
||||||
|
MAXTHREADS = 41
|
||||||
|
DATABASE_PATH = r"C:\\Database\\chunks\\"
|
||||||
|
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
CHUNK = "_chunk.txt"
|
||||||
|
_sentinel = object()
|
||||||
|
WORK_Q_SIZE = 35500000
|
||||||
|
|
||||||
|
|
||||||
|
def producer(out_q, control_q, filename, _logger):
|
||||||
|
"""
|
||||||
|
Produces items from the output queue and puts them into the control queue.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
out_q (Queue): Output queue.
|
||||||
|
control_q (Queue): Control queue.
|
||||||
|
filename (str): Name of the file.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
with open(DATABASE_PATH + filename, "r+", encoding=ENCODING) as operated_file:
|
||||||
|
print(f"Worker {filename} ")
|
||||||
|
line_no = 0
|
||||||
|
while True:
|
||||||
|
line = operated_file.readline()
|
||||||
|
line_no += 1
|
||||||
|
print(f"\t \t \t \t \t \t W{filename}{line_no}\r", end="")
|
||||||
|
|
||||||
|
if not line:
|
||||||
|
print(f"EOF {filename}")
|
||||||
|
break
|
||||||
|
|
||||||
|
if not line:
|
||||||
|
break
|
||||||
|
out_q.put(line)
|
||||||
|
try:
|
||||||
|
check = control_q.get(block=False)
|
||||||
|
except Empty:
|
||||||
|
check = False
|
||||||
|
|
||||||
|
if check is _sentinel:
|
||||||
|
print("TERM signal received")
|
||||||
|
control_q.put(check)
|
||||||
|
break
|
||||||
|
print(f"Worker finished: {filename}")
|
||||||
|
out_q.put(_sentinel)
|
||||||
|
|
||||||
|
|
||||||
|
# IMPORTANT!!! ONLY ONE CONSUMER THREAD AS WE ARE NOT PUTTING SENTINELS BACK
|
||||||
|
def consumer(in_q, control_q, doi, live_results, result_list, control_dict, no, _logger):
|
||||||
|
"""
|
||||||
|
Consumes items from an input queue and checks if DOI exists in the live results list.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
in_q (Queue): Input queue.
|
||||||
|
control_q (Queue): Control queue.
|
||||||
|
doi (list): List of DOI to search for.
|
||||||
|
live_results (list): List to store the search results.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
print(f"Consumer thread started: {no} no")
|
||||||
|
empty_counter = 0
|
||||||
|
alive_no = 0
|
||||||
|
while True:
|
||||||
|
done_check = True
|
||||||
|
try:
|
||||||
|
data = in_q.get(block=True, timeout = 1)
|
||||||
|
if data is _sentinel:
|
||||||
|
control_dict["sentinels"] += 1
|
||||||
|
print(f"Workers finished: {control_dict['sentinels']} reported by consumer {no}")
|
||||||
|
|
||||||
|
else:
|
||||||
|
empty_counter = 0
|
||||||
|
alive_no += 1
|
||||||
|
print(f"C{no}__{alive_no}\r", end="")
|
||||||
|
|
||||||
|
for item in result_list:
|
||||||
|
if item["DOI"] in data and not item["exists"]:
|
||||||
|
print(f"HIT in {no} content {data[0]} line {data[1]} file {data[2]} {item["exists"]}")
|
||||||
|
_logger.info(data)
|
||||||
|
_logger.info("HIT")
|
||||||
|
item["exists"] = True
|
||||||
|
live_results.append(item)
|
||||||
|
done_check = done_check and item["exists"]
|
||||||
|
if done_check:
|
||||||
|
control_q.put(_sentinel)
|
||||||
|
except Empty:
|
||||||
|
empty_counter += 1
|
||||||
|
time.sleep(1)
|
||||||
|
print(f"Consumer {no} empty")
|
||||||
|
if empty_counter > 5:
|
||||||
|
print("Consumer %s empty lvl 2", no)
|
||||||
|
time.sleep(2)
|
||||||
|
elif empty_counter > 10:
|
||||||
|
print(f"Consumer thread finished {no}")
|
||||||
|
break
|
||||||
|
|
||||||
|
if control_dict["sentinels"] >= MAXTHREADS:
|
||||||
|
_logger.info(f"All workers finished {no}")
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def search_for_doi(doi, live_results, _logger):
|
||||||
|
"""
|
||||||
|
Search for DOI in live_results using _logger for logging.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
doi (list): List of DOI to search for.
|
||||||
|
live_results (list): List to store the search results.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
control_dict = {"sentinels":0}
|
||||||
|
result_list = []
|
||||||
|
threads = []
|
||||||
|
work_q = Queue(maxsize=WORK_Q_SIZE)
|
||||||
|
control_q = Queue()
|
||||||
|
|
||||||
|
for item in doi:
|
||||||
|
result_list.append({"DOI": item[0], "exists": False, "data": item[1]})
|
||||||
|
|
||||||
|
for i in range (0, (len(doi)//1000)+2):
|
||||||
|
t_cons = Thread(
|
||||||
|
target=consumer, args=(work_q, control_q, doi, live_results, result_list, control_dict, i, _logger)
|
||||||
|
)
|
||||||
|
_logger.info("Consumer thread created")
|
||||||
|
threads.append(t_cons)
|
||||||
|
for i in range(0, MAXTHREADS):
|
||||||
|
# TEST DATA
|
||||||
|
# filename = "test" + str(i) + CHUNk
|
||||||
|
# DEPLOYMENT DATA
|
||||||
|
filename = str(i) + CHUNK
|
||||||
|
_logger.info(f"Creating worker thread no: {i}")
|
||||||
|
threads.append(
|
||||||
|
Thread(target=producer, args=(work_q, control_q, filename, _logger))
|
||||||
|
)
|
||||||
|
for worker in threads:
|
||||||
|
worker.start()
|
||||||
|
for worker in threads:
|
||||||
|
worker.join()
|
||||||
|
return result_list
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel("DEBUG")
|
||||||
|
h1 = logging.StreamHandler()
|
||||||
|
logger.addHandler(h1)
|
||||||
|
logger.info("TEST RUN")
|
||||||
|
live_result = []
|
||||||
|
logger.info(
|
||||||
|
search_for_doi(
|
||||||
|
[
|
||||||
|
("10.1002/9781118786352.wbieg0998.pub2", "DATA"),
|
||||||
|
("10.1002/j.2050-0416.2002.tb00563.x", "DATA"),
|
||||||
|
("10.1111/j.1365-2958.1994.tb00448.x", "DATA"),
|
||||||
|
("10.2165/00128415-200309690-00017", "DATA"),
|
||||||
|
("10.5772/48313", "DATA"),
|
||||||
|
("10.15803/ijnc.7.2_419", "DATA"),
|
||||||
|
("10.1051/0004-6361/201321596e", "DATA"),
|
||||||
|
("10.2307/40835941", "DATA"),
|
||||||
|
],
|
||||||
|
live_result,
|
||||||
|
logger,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
logger.info(live_result)
|
||||||
@@ -0,0 +1,206 @@
|
|||||||
|
"""
|
||||||
|
This module contains functions for searching for DOI (Digital Object Identifier)
|
||||||
|
in a list of live results.
|
||||||
|
It includes a producer-consumer pattern implementation using threads and queues.
|
||||||
|
|
||||||
|
Functions:
|
||||||
|
- producer: Reads lines from a file and puts them into an output queue.
|
||||||
|
- consumer: Consumes items from an input queue and checks if DOI exists in the live results list.
|
||||||
|
- search_for_doi: Searches for DOI in live results using the producer-consumer pattern.
|
||||||
|
|
||||||
|
Global Variables:
|
||||||
|
- MAXTHREADS: Maximum number of worker threads.
|
||||||
|
- DATABASE_PATH: Path to the database files.
|
||||||
|
- ENCODING: Encoding of the database files.
|
||||||
|
- CHUNK: File name pattern for the database files.
|
||||||
|
- _sentinel: Sentinel object used to signal termination.
|
||||||
|
- result_list: List to store the search results.
|
||||||
|
- WORK_Q_SIZE: Maximum size of the work queue.
|
||||||
|
"""
|
||||||
|
|
||||||
|
# TODO: Wpiemdolić to wszystko w klasę z loggerem przysłanym z góry
|
||||||
|
from multiprocessing import Manager, Process, Queue
|
||||||
|
from queue import Empty, Full
|
||||||
|
import time
|
||||||
|
|
||||||
|
# DATA FOR TEST ONLY
|
||||||
|
#MAXTHREADS = 6
|
||||||
|
#DATABASE_PATH = r'C:\\Database\\chunks_1\\'
|
||||||
|
|
||||||
|
# DEPLOYMENT DATA
|
||||||
|
MAXTHREADS = 41
|
||||||
|
DATABASE_PATH = r"C:\\Database\\chunks\\"
|
||||||
|
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
CHUNK = "_chunk.txt"
|
||||||
|
_sentinel = None
|
||||||
|
WORK_Q_SIZE = 355000
|
||||||
|
|
||||||
|
|
||||||
|
def producer(out_q, control_q, filename, _logger):
|
||||||
|
"""
|
||||||
|
Produces items from the output queue and puts them into the control queue.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
out_q (Queue): Output queue.
|
||||||
|
control_q (Queue): Control queue.
|
||||||
|
filename (str): Name of the file.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
with open(DATABASE_PATH + filename, "r+", encoding=ENCODING) as operated_file:
|
||||||
|
print(f"Worker {filename} ")
|
||||||
|
line_no = 0
|
||||||
|
while True:
|
||||||
|
line = operated_file.readline()
|
||||||
|
line_no += 1
|
||||||
|
print(f"\t \t \t W{filename}{line_no}\r", end="")
|
||||||
|
|
||||||
|
if not line:
|
||||||
|
print(f"EOF {filename}")
|
||||||
|
break
|
||||||
|
try:
|
||||||
|
out_q.put([line, line_no, filename], block = True, timeout = 60)
|
||||||
|
except Full:
|
||||||
|
print("queue fulll")
|
||||||
|
try:
|
||||||
|
check = control_q.get(block=False)
|
||||||
|
except Empty:
|
||||||
|
check = False
|
||||||
|
if check is None:
|
||||||
|
print(f"TERM signal received {filename}")
|
||||||
|
control_q.put(check)
|
||||||
|
control_q.close()
|
||||||
|
break
|
||||||
|
print(f"Worker finished: {filename}")
|
||||||
|
out_q.put(_sentinel)
|
||||||
|
out_q.close()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# IMPORTANT!!! ONLY ONE CONSUMER THREAD AS WE ARE NOT PUTTING SENTINELS BACK
|
||||||
|
def consumer(in_q, control_q, doi, live_results, result_list, control_dict, no, _logger):
|
||||||
|
"""
|
||||||
|
Consumes items from an input queue and checks if DOI exists in the live results list.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
in_q (Queue): Input queue.
|
||||||
|
control_q (Queue): Control queue.
|
||||||
|
doi (list): List of DOI to search for.
|
||||||
|
live_results (list): List to store the search results.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
print(f"Consumer thread started: {no} no")
|
||||||
|
empty_counter = 0
|
||||||
|
alive_no = 0
|
||||||
|
while True:
|
||||||
|
done_check = True
|
||||||
|
try:
|
||||||
|
data = in_q.get(block=True, timeout = 1)
|
||||||
|
empty_counter = 0
|
||||||
|
if data is None:
|
||||||
|
control_dict["sentinels"] += 1
|
||||||
|
print(f"Workers finished: {control_dict['sentinels']} reported by consumer {no}")
|
||||||
|
else:
|
||||||
|
empty_counter = 0
|
||||||
|
if alive_no > 9:
|
||||||
|
alive_no = 0
|
||||||
|
alive_no += 1
|
||||||
|
print(f"C{no}{alive_no}\r", end="")
|
||||||
|
for item in result_list:
|
||||||
|
if item["DOI"] in data[0] and not item["exists"]:
|
||||||
|
print(f"HIT in {no} content {data[0]} line {data[1]} file {data[2]} {item["exists"]}")
|
||||||
|
item["exists"] = True
|
||||||
|
live_results.append(item)
|
||||||
|
done_check = done_check and item["exists"]
|
||||||
|
if done_check:
|
||||||
|
control_q.put(_sentinel)
|
||||||
|
except Empty:
|
||||||
|
empty_counter += 1
|
||||||
|
time.sleep(1)
|
||||||
|
print(f"Consumer {no} empty")
|
||||||
|
if empty_counter > 5:
|
||||||
|
print("Consumer %s empty lvl 2", no)
|
||||||
|
time.sleep(2)
|
||||||
|
elif empty_counter > 10:
|
||||||
|
print(f"Consumer thread finished {no}")
|
||||||
|
break
|
||||||
|
if control_dict["sentinels"] >= MAXTHREADS:
|
||||||
|
print(f"All workers finished killing {no}")
|
||||||
|
break
|
||||||
|
print("Consumer thread finished %s",no )
|
||||||
|
control_q.close()
|
||||||
|
|
||||||
|
|
||||||
|
def search_for_doi(doi, live_results, _logger):
|
||||||
|
"""
|
||||||
|
Search for DOI in live_results using _logger for logging.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
doi (list): List of DOI to search for.
|
||||||
|
live_results (list): List to store the search results.
|
||||||
|
_logger: Logger object for logging.
|
||||||
|
"""
|
||||||
|
with Manager() as manager:
|
||||||
|
control_dict = manager.dict()
|
||||||
|
control_dict["sentinels"] = 0
|
||||||
|
result_list = []
|
||||||
|
for item in doi:
|
||||||
|
result_list.append({"DOI": item[0], "exists": False, "data": item[1]})
|
||||||
|
result_list_proxy = manager.list(result_list)
|
||||||
|
|
||||||
|
threads = []
|
||||||
|
work_q = Queue(maxsize=WORK_Q_SIZE)
|
||||||
|
control_q = Queue()
|
||||||
|
for i in range(0, MAXTHREADS):
|
||||||
|
# TEST DATA
|
||||||
|
#filename = "test" + str(i) + CHUNK
|
||||||
|
# DEPLOYMENT DATA
|
||||||
|
filename = str(i) + CHUNK
|
||||||
|
print(f"Creating worker thread no: {i}")
|
||||||
|
threads.append(
|
||||||
|
Process(target=producer, args=(work_q, control_q, filename, _logger))
|
||||||
|
)
|
||||||
|
for i in range (0, (len(doi)//1000)+5):
|
||||||
|
t_cons = Process(
|
||||||
|
target=consumer, args=(work_q, control_q, doi, live_results, result_list_proxy, control_dict, i, _logger)
|
||||||
|
)
|
||||||
|
print("Consumer thread created")
|
||||||
|
threads.append(t_cons)
|
||||||
|
|
||||||
|
for worker in threads:
|
||||||
|
worker.start()
|
||||||
|
for worker in threads:
|
||||||
|
worker.join()
|
||||||
|
result_list = list(result_list_proxy)
|
||||||
|
for worker in threads:
|
||||||
|
worker.close()
|
||||||
|
return result_list
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel("DEBUG")
|
||||||
|
h1 = logging.StreamHandler()
|
||||||
|
logger.addHandler(h1)
|
||||||
|
print("TEST RUN")
|
||||||
|
live_result = []
|
||||||
|
print(
|
||||||
|
search_for_doi(
|
||||||
|
[
|
||||||
|
("10.1002/9781118786352.wbieg0998.pub2", "DATA"),
|
||||||
|
("10.1002/j.2050-0416.2002.tb00563.x", "DATA"),
|
||||||
|
("10.1111/j.1365-2958.1994.tb00448.x", "DATA"),
|
||||||
|
("10.2165/00128415-200309690-00017", "DATA"),
|
||||||
|
("10.5772/48313", "DATA"),
|
||||||
|
("10.15803/ijnc.7.2_419", "DATA"),
|
||||||
|
("10.1051/0004-6361/201321596e", "DATA"),
|
||||||
|
("10.2307/40835941", "DATA"),
|
||||||
|
("NOT_ON_LIST", "DATA"),
|
||||||
|
],
|
||||||
|
live_result,
|
||||||
|
logger,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
print(live_result)
|
||||||
@@ -0,0 +1,45 @@
|
|||||||
|
import asyncio
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
from json.decoder import JSONDecodeError
|
||||||
|
from logging import handlers
|
||||||
|
import requests
|
||||||
|
|
||||||
|
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
||||||
|
SEND_RESULTS = "/conjurer"
|
||||||
|
|
||||||
|
async def send_results():
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel(logging.DEBUG)
|
||||||
|
h1 = handlers.RotatingFileHandler(
|
||||||
|
filename="D:\\logs\\librarian.log",
|
||||||
|
encoding="utf-8",
|
||||||
|
mode="a",
|
||||||
|
maxBytes=6 * 1024 * 1024,
|
||||||
|
backupCount=6,
|
||||||
|
)
|
||||||
|
|
||||||
|
logger.addHandler(h1)
|
||||||
|
|
||||||
|
with open("s_results.json", "r+", encoding="utf-8") as s_file:
|
||||||
|
try:
|
||||||
|
database = json.load(s_file)
|
||||||
|
except JSONDecodeError:
|
||||||
|
pass
|
||||||
|
result = database["4bacbca0-ce3b-45c6-90a1-05c70b46b740"]
|
||||||
|
result = {"4bacbca0-ce3b-45c6-90a1-05c70b46b740":result}
|
||||||
|
|
||||||
|
coroutine = asyncio.to_thread(
|
||||||
|
requests.post,
|
||||||
|
f"{MAIN_BOT_ADDRESS}{SEND_RESULTS}",
|
||||||
|
json=result,
|
||||||
|
timeout=360,
|
||||||
|
)
|
||||||
|
logger.info("SENT")
|
||||||
|
result = await coroutine
|
||||||
|
logger.info(result.status_code)
|
||||||
|
logger.info("SEND CONFIRMED")
|
||||||
|
async def main():
|
||||||
|
await send_results()
|
||||||
|
if __name__ == "__main__":
|
||||||
|
asyncio.run(main())
|
||||||
@@ -0,0 +1,553 @@
|
|||||||
|
# This Python file uses the following encoding: utf-8
|
||||||
|
"""
|
||||||
|
The provided Python script sets up a Flask web server to manage a list of music files, with
|
||||||
|
functions for rescanning the music folder, updating the music list, and serving the music list via
|
||||||
|
API endpoints.
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import requests
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
from datetime import datetime
|
||||||
|
from logging import handlers
|
||||||
|
from pathlib import Path
|
||||||
|
from platform import uname
|
||||||
|
from sys import platform
|
||||||
|
from flask import Flask, jsonify, redirect, request, send_from_directory, render_template
|
||||||
|
from waitress import serve
|
||||||
|
|
||||||
|
MAIN_BOT_ADDRESS = "http://192.168.1.191:5000"
|
||||||
|
MUSIC_TRACKER = "/prepped_tracks"
|
||||||
|
HOST_ADDRESS = "192.168.1.15"
|
||||||
|
PORT_ADDRESS = 5000
|
||||||
|
if platform in ("linux", "linux2"):
|
||||||
|
SEPARATOR_FILE_PATH = "/"
|
||||||
|
if "microsoft-standard" in uname().release:
|
||||||
|
LOGFILE = "/home/mtuszowski/conjurer/discord_mus_service.log"
|
||||||
|
MUSIC_FOLDER = "/mnt/g/Muzyka/"
|
||||||
|
NETRC_FILE = "/home/mtuszowski/.netrc"
|
||||||
|
LOGSTORE = "/home/mtuszowski/conjurer/logs/"
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
RADIOLOG_PATH = '/home/pi/Conjurer/radio_log.log'
|
||||||
|
|
||||||
|
else:
|
||||||
|
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
||||||
|
NETRC_FILE = "/home/pi/.netrc"
|
||||||
|
LOGSTORE = "/home/pi/RetroPie/logs/"
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
||||||
|
PRIORITY_FOLDER = "/home/pi/RetroPie/mp3/Magiczne i chuj/"
|
||||||
|
RADIOLOG_PATH = '/home/pi/Conjurer/radio_log.log'
|
||||||
|
|
||||||
|
|
||||||
|
music_file_list = []
|
||||||
|
priority_list = []
|
||||||
|
|
||||||
|
def rescan():
|
||||||
|
"""
|
||||||
|
The `rescan` function logs a message, scans for mp3 files in a specified folder,
|
||||||
|
and adds them to a list of music files.
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
logger.info("Rescan triggered")
|
||||||
|
|
||||||
|
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
||||||
|
temp_music_file = mp3_item.as_posix()
|
||||||
|
if platform == "win32":
|
||||||
|
temp_music_file = temp_music_file.replace("/", "\\")
|
||||||
|
music_file_list.append(temp_music_file)
|
||||||
|
|
||||||
|
for mp3_item in Path.glob(Path(PRIORITY_FOLDER), "**/*.mp3"):
|
||||||
|
temp_music_file = mp3_item.as_posix()
|
||||||
|
if platform == "win32":
|
||||||
|
temp_music_file = temp_music_file.replace("/", "\\")
|
||||||
|
priority_list.append(temp_music_file)
|
||||||
|
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/all_playlist.playlist", "w", encoding="utf-8"
|
||||||
|
) as w_file:
|
||||||
|
try:
|
||||||
|
for item in music_file_list:
|
||||||
|
w_file.write(item)
|
||||||
|
w_file.write("\n")
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
pass
|
||||||
|
with open("/home/pi/Conjurer/hit.playlist", "w", encoding="utf-8") as w_file:
|
||||||
|
try:
|
||||||
|
for item in priority_list:
|
||||||
|
w_file.write(item)
|
||||||
|
w_file.write("\n")
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def thread_rescan():
|
||||||
|
"""
|
||||||
|
The `thread_rescan` function periodically triggers a rescan operation after a specified time
|
||||||
|
interval.
|
||||||
|
"""
|
||||||
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
logger.info("Starting filesystemupdater")
|
||||||
|
while True:
|
||||||
|
time.sleep(60 * 60 * 24)
|
||||||
|
logger.info("Rescan triggered")
|
||||||
|
rescan()
|
||||||
|
|
||||||
|
|
||||||
|
def scan_tracks():
|
||||||
|
#Set the filename and open the file
|
||||||
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
|
||||||
|
file = open(RADIOLOG_PATH,'r')
|
||||||
|
|
||||||
|
#Find the size of the file and move to the end
|
||||||
|
st_results = os.stat(RADIOLOG_PATH)
|
||||||
|
st_size = st_results[6]
|
||||||
|
file.seek(st_size)
|
||||||
|
|
||||||
|
while 1:
|
||||||
|
where = file.tell()
|
||||||
|
line = file.readline()
|
||||||
|
if not line:
|
||||||
|
time.sleep(1)
|
||||||
|
file.seek(where)
|
||||||
|
else:
|
||||||
|
if re.match(".*Prepared.*",line):
|
||||||
|
result = None
|
||||||
|
if re.match(".*jingles.*", line):
|
||||||
|
logger.info("jingles")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["jingles", line]
|
||||||
|
elif re.match(".*priority.*", line):
|
||||||
|
logger.info("priority")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["priority", line]
|
||||||
|
elif re.match(".*hit.*", line):
|
||||||
|
logger.info("priority")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["hit", line]
|
||||||
|
elif re.match(".*all_playlist.*", line):
|
||||||
|
logger.info("all")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["all", line]
|
||||||
|
elif re.match(".*request.*", line):
|
||||||
|
logger.info("requests")
|
||||||
|
logger.info(line) # already has newline
|
||||||
|
result = ["requests", line]
|
||||||
|
if result:
|
||||||
|
returned = requests.post(
|
||||||
|
f"{MAIN_BOT_ADDRESS}{MUSIC_TRACKER}",
|
||||||
|
json=result,
|
||||||
|
timeout=360)
|
||||||
|
logger.info("SENT")
|
||||||
|
logger.info(returned.status_code)
|
||||||
|
logger.info("SEND CONFIRMED")
|
||||||
|
time.sleep(0.1)
|
||||||
|
|
||||||
|
|
||||||
|
app = Flask(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: Odpalić wyszukiwarki w wątkach i dopiero po wszystkim zsumować wyszukiwanie.
|
||||||
|
def wyszukaj(word_list, how_many, _logger=None, return_to_bot=True):
|
||||||
|
"""
|
||||||
|
Searches for music files based on a list of keywords and returns a list of matching files.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
word_list (list): A list of keywords to search for.
|
||||||
|
how_many (int): The maximum number of matching files to return.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: A list of matching files, sorted by their search weight.
|
||||||
|
|
||||||
|
"""
|
||||||
|
if _logger is None:
|
||||||
|
fun_logger = logging.getLogger()
|
||||||
|
else:
|
||||||
|
fun_logger = _logger
|
||||||
|
return_list = []
|
||||||
|
search_weight = []
|
||||||
|
|
||||||
|
for _ in range(len(music_file_list)):
|
||||||
|
search_weight.append((0, ""))
|
||||||
|
|
||||||
|
time_start = datetime.now()
|
||||||
|
if int(how_many) > 0:
|
||||||
|
skip_start = 2
|
||||||
|
else:
|
||||||
|
skip_start = 1
|
||||||
|
for word in word_list[skip_start:]:
|
||||||
|
token_weight = len(word)
|
||||||
|
fun_logger.info("Słowo kluczowe: %s", word)
|
||||||
|
itr = 0
|
||||||
|
for file in music_file_list:
|
||||||
|
file = file.split(SEPARATOR_FILE_PATH)
|
||||||
|
|
||||||
|
char_remove = [
|
||||||
|
".",
|
||||||
|
"^",
|
||||||
|
"$",
|
||||||
|
"*",
|
||||||
|
"+",
|
||||||
|
"?",
|
||||||
|
"{",
|
||||||
|
"}",
|
||||||
|
"[",
|
||||||
|
"]",
|
||||||
|
"\\",
|
||||||
|
"/",
|
||||||
|
"|",
|
||||||
|
"(",
|
||||||
|
")",
|
||||||
|
"!",
|
||||||
|
",",
|
||||||
|
"-",
|
||||||
|
":",
|
||||||
|
"mp3",
|
||||||
|
]
|
||||||
|
all_words = []
|
||||||
|
for f_iter in file:
|
||||||
|
for char in char_remove:
|
||||||
|
f_iter = remove_characters(f_iter, char)
|
||||||
|
tmp = f_iter.split()
|
||||||
|
all_words.extend(tmp)
|
||||||
|
pingu = 1
|
||||||
|
pattern_len = len(all_words)
|
||||||
|
if platform == "win32":
|
||||||
|
skip = 2
|
||||||
|
else:
|
||||||
|
skip = 4
|
||||||
|
matched_times = 1
|
||||||
|
# TODO: To do wątków albo for asynchroniczny
|
||||||
|
for itm in all_words[skip:]:
|
||||||
|
pingu += 1
|
||||||
|
pattern = ".*" + word + ".*"
|
||||||
|
if re.match(pattern, itm, re.IGNORECASE):
|
||||||
|
temp_weight = (
|
||||||
|
search_weight[itr][0]
|
||||||
|
+ (token_weight + (pingu**1.5) / pattern_len) / matched_times
|
||||||
|
)
|
||||||
|
search_weight[itr] = (
|
||||||
|
temp_weight,
|
||||||
|
music_file_list[itr],
|
||||||
|
)
|
||||||
|
matched_times += 1
|
||||||
|
itr += 1
|
||||||
|
|
||||||
|
fun_logger.info(
|
||||||
|
"Stworzylem tablice wag zajęło mi to %s", datetime.now() - time_start
|
||||||
|
)
|
||||||
|
time_start = datetime.now()
|
||||||
|
item_to_search = max_weight(search_weight)
|
||||||
|
itr = 0
|
||||||
|
if item_to_search == 0:
|
||||||
|
return []
|
||||||
|
not_found = True
|
||||||
|
return_list = []
|
||||||
|
if int(how_many) <= 0:
|
||||||
|
fun_logger.info("Jeden plik do zagrania")
|
||||||
|
while not_found:
|
||||||
|
if search_weight[itr][0] == item_to_search:
|
||||||
|
return_list.append(search_weight[itr])
|
||||||
|
if not return_to_bot:
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/priority_queue.playlist",
|
||||||
|
"r+",
|
||||||
|
encoding="utf-8",
|
||||||
|
) as s_file:
|
||||||
|
s_file.write(search_weight[itr][1])
|
||||||
|
break
|
||||||
|
itr += 1
|
||||||
|
else:
|
||||||
|
fun_logger.info("Wiele plików do zagrania")
|
||||||
|
search_weight.sort(key=lambda x: x[0], reverse=True)
|
||||||
|
return_list.extend(search_weight[: int(how_many)])
|
||||||
|
fun_logger.info("Done: %s", return_list)
|
||||||
|
return return_list
|
||||||
|
|
||||||
|
|
||||||
|
def max_weight(lista):
|
||||||
|
"""
|
||||||
|
Take a list of weights and return the maximum weight.
|
||||||
|
|
||||||
|
:param lista: It seems like the parameter `lista` is a list of items, possibly representing weights.
|
||||||
|
The function name `max_weight` suggests that the function is intended to find the maximum weight
|
||||||
|
from the list. However, without more context or information about the problem, it's difficult to say
|
||||||
|
for sure what the
|
||||||
|
"""
|
||||||
|
maximum_weight = 0
|
||||||
|
for iterator in lista:
|
||||||
|
if iterator[0] > maximum_weight:
|
||||||
|
maximum_weight = iterator[0]
|
||||||
|
return maximum_weight
|
||||||
|
|
||||||
|
|
||||||
|
def remove_characters(string, character):
|
||||||
|
"""
|
||||||
|
The `remove_characters` function removes all occurrences of a specified character from a given
|
||||||
|
string.
|
||||||
|
|
||||||
|
:param string: The string parameter is the input string from which characters will be removed
|
||||||
|
:param character: The character parameter is the character that you want to remove from the string
|
||||||
|
:return: a new string where all occurrences of the specified character have been removed.
|
||||||
|
"""
|
||||||
|
return string.replace(character, "")
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/stream", methods=["GET"])
|
||||||
|
def stream_music():
|
||||||
|
"""
|
||||||
|
The function `stream_music` is a route handler for the "/stream" endpoint.
|
||||||
|
It returns a JSON response containing a key "music_file_list" with the value of the variable `music_file_list`.
|
||||||
|
|
||||||
|
:return: A JSON response containing a key "music_file_list" with the value of the variable `music_file_list`.
|
||||||
|
"""
|
||||||
|
|
||||||
|
#return send_from_directory("/tmp/hls", "stream.m3u8")
|
||||||
|
return render_template("/home/pi/Conjurer/stream.html")
|
||||||
|
|
||||||
|
@app.route("/<string:file_name>")
|
||||||
|
def stream(file_name):
|
||||||
|
"""
|
||||||
|
Stream the specified file from the video directory.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
file_name (str): The name of the file to be streamed.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Response: The response containing the streamed file.
|
||||||
|
"""
|
||||||
|
# trunk-ignore(bandit/B108)
|
||||||
|
video_dir = "/tmp/hls"
|
||||||
|
return send_from_directory(video_dir, file_name)
|
||||||
|
|
||||||
|
@app.route("/stream_mp3", methods=["GET"])
|
||||||
|
def stream_music_mp3():
|
||||||
|
"""
|
||||||
|
The function `stream_music_mp3` is a route handler for the "/stream_mp3" endpoint.
|
||||||
|
It redirects the user to the URL "http://www.example.com" with a status code of 302.
|
||||||
|
|
||||||
|
:return: A redirect response to "http://www.example.com" with a status code of 302.
|
||||||
|
"""
|
||||||
|
return redirect("http://www.example.com", code=302)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/clear_pr_pls", methods=["GET"])
|
||||||
|
def clear_pr_pls():
|
||||||
|
"""
|
||||||
|
The function `clear_pr_pls` clears the contents of the priority queue playlist file.
|
||||||
|
|
||||||
|
:return: A JSON response indicating the success of the operation.
|
||||||
|
"""
|
||||||
|
app.logger.info("CLEARING PLAYLIST")
|
||||||
|
with open('/home/pi/Conjurer/priority_queue.playlist', 'w', encoding='utf-8') as cleared_pl:
|
||||||
|
cleared_pl.write('')
|
||||||
|
|
||||||
|
return_data = jsonify(isError=False, message="Success", statusCode=200, data=[])
|
||||||
|
return return_data, 200
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/mp3", methods=["GET"])
|
||||||
|
def get_music_list():
|
||||||
|
"""
|
||||||
|
The function `get_music_list` returns a JSON object containing a list of music files.
|
||||||
|
:return: A JSON response containing a key "music_file_list" with the value of the variable
|
||||||
|
`music_file_list`.
|
||||||
|
"""
|
||||||
|
return jsonify({"music_file_list": music_file_list})
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/update_mp3", methods=["POST"])
|
||||||
|
def update_music_list():
|
||||||
|
"""
|
||||||
|
The function `update_music_list` receives a POST request with a JSON payload, logs the received
|
||||||
|
item, adds it to a music file list, and returns a success message along with the updated record.
|
||||||
|
:return: The function `update_music_list` is returning a tuple containing a JSON response and a
|
||||||
|
status code. The JSON response includes keys `isError`, `message`, `statusCode`, and `data`,
|
||||||
|
with values indicating the success of the operation and the data that was
|
||||||
|
received and added to the `music_file_list`.
|
||||||
|
The status code returned is 200, indicating a successful response.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record["item"])
|
||||||
|
music_file_list.append(record["item"])
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data=record),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/get_music", methods=["POST"])
|
||||||
|
def look_for_playlist():
|
||||||
|
"""
|
||||||
|
The function `look_for_playlist` receives a POST request with a JSON payload, logs the received
|
||||||
|
item, adds it to a music file list, and returns a success message along with the updated record.
|
||||||
|
|
||||||
|
:return: A tuple containing a JSON response and a status code. The JSON response includes keys `isError`,
|
||||||
|
`message`, `statusCode`, and `data`, with values indicating the success of the operation and the
|
||||||
|
data that was received and added to the `music_file_list`. The status code returned is 200,
|
||||||
|
indicating a successful response.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["lista_slow"])
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
app.logger.info(record["dlugosc_plejlisty"])
|
||||||
|
return_data = wyszukaj(
|
||||||
|
record["lista_slow"], record["dlugosc_plejlisty"], app.logger, True
|
||||||
|
)
|
||||||
|
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data=return_data),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/request_radio_file", methods=["POST"])
|
||||||
|
def add_request():
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["lista_slow"])
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
return_data = wyszukaj(
|
||||||
|
record["lista_slow"], 0, app.logger, False
|
||||||
|
)
|
||||||
|
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/request.playlist", "a", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
for item in return_data:
|
||||||
|
s_file.write(item[1] + "\n")
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data={"status": "OK"}),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
@app.route("/create_priority_playlist", methods=["POST"])
|
||||||
|
def create_priority_playlist():
|
||||||
|
"""
|
||||||
|
The function `create_priority_playlist` receives a POST request with a JSON payload, logs the received
|
||||||
|
item, adds it to a music file list, and returns a success message along with the updated record.
|
||||||
|
|
||||||
|
:return: A tuple containing a JSON response and a status code.
|
||||||
|
The JSON response includes keys `isError`,
|
||||||
|
`message`, `statusCode`, and `data`, with values
|
||||||
|
indicating the success of the operation and the
|
||||||
|
data that was received and added to the `music_file_list`.
|
||||||
|
The status code returned is 200,indicating a successful response.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["lista_slow"])
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
app.logger.info(record["dlugosc_plejlisty"])
|
||||||
|
return_data = wyszukaj(
|
||||||
|
record["lista_slow"], record["dlugosc_plejlisty"], app.logger, False
|
||||||
|
)
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/request.playlist", "a", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
for item in return_data:
|
||||||
|
s_file.write(item[1] + "\n")
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data={"status": "OK"}),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/add_to_priority", methods=["POST"])
|
||||||
|
def add_to_priority():
|
||||||
|
"""
|
||||||
|
The function `add_to_priority` receives a POST request with a JSON payload, logs the received
|
||||||
|
item, adds it to a music file list, and returns a success message along with the updated record.
|
||||||
|
|
||||||
|
:return: A tuple containing a JSON response and a status code.
|
||||||
|
The JSON response includes keys `isError`,
|
||||||
|
`message`, `statusCode`, and `data`, with values
|
||||||
|
indicating the success of the operation and the
|
||||||
|
data that was received and added to the `music_file_list`.
|
||||||
|
The status code returned is 200,indicating a successful response.
|
||||||
|
"""
|
||||||
|
record = json.loads(request.data)
|
||||||
|
app.logger.info(record)
|
||||||
|
app.logger.info(record["lista_slow"])
|
||||||
|
app.logger.info(record["UUID"])
|
||||||
|
app.logger.info(record["dlugosc_plejlisty"])
|
||||||
|
return_data = wyszukaj(
|
||||||
|
record["lista_slow"], record["dlugosc_plejlisty"], app.logger, False
|
||||||
|
)
|
||||||
|
with open(
|
||||||
|
"/home/pi/Conjurer/priority_queue.playlist", "a", encoding="utf-8"
|
||||||
|
) as s_file:
|
||||||
|
for item in return_data:
|
||||||
|
s_file.write(item[1] + "\n")
|
||||||
|
return_data = (
|
||||||
|
jsonify(isError=False, message="Success", statusCode=200, data={"status": "OK"}),
|
||||||
|
200,
|
||||||
|
)
|
||||||
|
return return_data
|
||||||
|
|
||||||
|
|
||||||
|
def flask_debug():
|
||||||
|
"""
|
||||||
|
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
||||||
|
Do not use for production for fucks sake.
|
||||||
|
"""
|
||||||
|
# trunk-ignore(bandit/B201)
|
||||||
|
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
|
def waitress_run():
|
||||||
|
"""
|
||||||
|
The `waitress_run` function serves the `app` on host HOST_ADDRESS
|
||||||
|
and port 5000 using the Waitress WSGI server.
|
||||||
|
|
||||||
|
This function starts the Waitress server and listens for incoming requests
|
||||||
|
on the specified host and port. It uses the `serve` function from the Waitress
|
||||||
|
library to handle the serving process.
|
||||||
|
|
||||||
|
Parameters:
|
||||||
|
None
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None
|
||||||
|
"""
|
||||||
|
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
logger = logging.getLogger("conjurer_musician")
|
||||||
|
logger.setLevel(logging.DEBUG)
|
||||||
|
formatter = logging.Formatter(
|
||||||
|
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||||
|
)
|
||||||
|
handler = handlers.RotatingFileHandler(
|
||||||
|
filename=LOGFILE,
|
||||||
|
encoding=ENCODING,
|
||||||
|
mode="a",
|
||||||
|
maxBytes=6 * 1024 * 1024,
|
||||||
|
backupCount=6,
|
||||||
|
)
|
||||||
|
handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(handler)
|
||||||
|
|
||||||
|
rescan()
|
||||||
|
logger.info("Started")
|
||||||
|
threads = []
|
||||||
|
# threads.append(threading.Thread(target=flask_debug))
|
||||||
|
threads.append(threading.Thread(target=waitress_run))
|
||||||
|
threads.append(threading.Thread(target=thread_rescan))
|
||||||
|
threads.append(threading.Thread(target=scan_tracks))
|
||||||
|
|
||||||
|
for worker in threads:
|
||||||
|
worker.start()
|
||||||
|
|
||||||
|
for worker in threads:
|
||||||
|
worker.join()
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Conjurer_musician
|
||||||
|
After=multi-user.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
ExecStart=/home/pi/Conjurer/env/bin/python3 /home/pi/Conjurer/conjurer_musician.py
|
||||||
|
Restart=on-abort
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
# This Python file uses the following encoding: utf-8
|
||||||
|
"""This module contains utility scripts used by musi radiostation Conjurer"""
|
||||||
|
import time
|
||||||
|
import re
|
||||||
|
|
||||||
|
NAME = "/home/pi/Conjurer/persistence.log"
|
||||||
|
|
||||||
|
def print_top():
|
||||||
|
"""
|
||||||
|
Prints the top 65 lines from a file specified by the NAME variable.
|
||||||
|
"""
|
||||||
|
while True:
|
||||||
|
line = ""
|
||||||
|
buffer = ""
|
||||||
|
with open(NAME, "r", encoding="utf-8") as f:
|
||||||
|
for _ in range (45):
|
||||||
|
line = f.readline()
|
||||||
|
if re.match(".*mp3", line):
|
||||||
|
buffer += line
|
||||||
|
print("=====================================")
|
||||||
|
print(buffer)
|
||||||
|
print("=====================================")
|
||||||
|
time.sleep(180)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
print_top()
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
$dodaj_do_ulubionych 100 nightwish
|
||||||
|
$dodaj_do_ulubionych 100 amon amarth
|
||||||
|
$dodaj_do_ulubionych 100 tactical sekt
|
||||||
|
$dodaj_do_ulubionych 100 lacuna coil
|
||||||
|
$dodaj_do_ulubionych 100 sefa
|
||||||
|
$dodaj_do_ulubionych 100 frenchcore
|
||||||
|
$dodaj_do_ulubionych 100 peacock
|
||||||
|
$dodaj_do_ulubionych 100 vavamuffin
|
||||||
|
$dodaj_do_ulubionych 100 abradab
|
||||||
|
$dodaj_do_ulubionych 100 eluveitie
|
||||||
|
$dodaj_do_ulubionych 100 youtube
|
||||||
|
|
||||||
|
$dodaj_do_ulubionych 100 elyose
|
||||||
|
$dodaj_do_ulubionych 1 artisans du chaos
|
||||||
Executable
+34804
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,167 @@
|
|||||||
|
# FILEPATH: /home/mtuszowski/conjurer/conjurer_musician/radio_conjurer.liq
|
||||||
|
|
||||||
|
# This script sets up a Liquidsoap radio stream with various features and configurations.
|
||||||
|
|
||||||
|
# Load icecast credentials from a JSON file
|
||||||
|
let json.parse credentials = file.contents("/home/pi/Conjurer/icecast_credentials.json")
|
||||||
|
|
||||||
|
# Enable replaygain metadata processing
|
||||||
|
enable_replaygain_metadata()
|
||||||
|
|
||||||
|
# Set up a playlog for tracking played tracks
|
||||||
|
|
||||||
|
l = playlog(duration = 72000.0, persistency="/home/pi/Conjurer/persistence.log")
|
||||||
|
|
||||||
|
# Function to check if a track can be played based on its metadata
|
||||||
|
def check(r)
|
||||||
|
m = request.metadata(r)
|
||||||
|
if l.last(m) < 36000. then
|
||||||
|
log.info("Rejecting #{m['filename']} (played #{l.last(m)}s ago).")
|
||||||
|
false
|
||||||
|
else
|
||||||
|
l.add(m)
|
||||||
|
true
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Define playlists to be used in the stream
|
||||||
|
s1 = replaygain(playlist(reload_mode="watch", check_next=check, "/home/pi/Conjurer/all_playlist.playlist"))
|
||||||
|
s2 = replaygain(playlist(reload_mode="watch", check_next=check, "/home/pi/Conjurer/priority_queue.playlist"))
|
||||||
|
s3 = replaygain(playlist(reload_mode="watch", check_next=check, "/home/pi/Conjurer/hit.playlist"))
|
||||||
|
|
||||||
|
# Create a request queue for user-generated requests
|
||||||
|
requests_queue = request.queue()
|
||||||
|
|
||||||
|
# Function to process the request queue and add new requests
|
||||||
|
def queue_processing()
|
||||||
|
text=file.lines("/home/pi/Conjurer/request.playlist")
|
||||||
|
if text != [] then
|
||||||
|
list.iter(fun(item) -> requests_queue.push.uri(item), text)
|
||||||
|
file.remove("/home/pi/Conjurer/request.playlist")
|
||||||
|
f = file.open("/home/pi/Conjurer/request.playlist", create=true)
|
||||||
|
f.close()
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Randomly select a playlist with weights
|
||||||
|
s4 = random(id="randomizer", weights=[2, 3, 5], [s1, s2, s3])
|
||||||
|
|
||||||
|
# Load jingles playlist
|
||||||
|
jingles = (playlist(reload_mode="watch", "/home/pi/Conjurer/jingles.playlist"))
|
||||||
|
|
||||||
|
# Create the main stream with random playlist and jingles
|
||||||
|
s = rotate(id="randomizer", weights=[10, 1], [s4, jingles])
|
||||||
|
|
||||||
|
s = crossfade(fade_out=2., fade_in=2., duration=4., fallback(id="switcher", track_sensitive=false, [requests_queue, s]))
|
||||||
|
|
||||||
|
# Set up an interactive harbor for controlling the stream
|
||||||
|
interactive.harbor(port = 9999)
|
||||||
|
|
||||||
|
# Set up interactive controls for bass boost
|
||||||
|
f = interactive.float("f", description="Frequency", min=0., max=1000., unit="Hz", 200.)
|
||||||
|
g = interactive.float("g", description="Gain", min=0., max=20., unit="dB", 8.)
|
||||||
|
b = bass_boost(frequency=f, gain=g, s)
|
||||||
|
s = add([s, b])
|
||||||
|
|
||||||
|
# Set up interactive control for main volume
|
||||||
|
a = interactive.float("main_volume", min=0., max=20., 1.)
|
||||||
|
s = compress.multiband.interactive(bands=7, s)
|
||||||
|
|
||||||
|
# Apply audio processing effects
|
||||||
|
s = nrj(normalize(s))
|
||||||
|
s = amplify(a, s)
|
||||||
|
|
||||||
|
mic = buffer(input.pulseaudio()) # Microphone
|
||||||
|
mic = gate(threshold=-30., range=-80., mic)
|
||||||
|
mic = blank.strip(max_blank=10., min_noise=.1, threshold=-20., mic)
|
||||||
|
|
||||||
|
s = add([s, mic])
|
||||||
|
#w = interactive.float("wetness", min=0., max=1., 1.)
|
||||||
|
#s = dry_wet(w,s,s2)
|
||||||
|
|
||||||
|
|
||||||
|
# Skip blank sections in the stream
|
||||||
|
s = blank.skip(max_blank=10., s)
|
||||||
|
|
||||||
|
# Configure logging settings
|
||||||
|
log_to_stdout = true
|
||||||
|
log_to_file = true
|
||||||
|
logpath = "/home/pi/Conjurer/radio_log.log"
|
||||||
|
loglevel = 3
|
||||||
|
set("log.stdout", log_to_stdout)
|
||||||
|
set("log.level", loglevel)
|
||||||
|
|
||||||
|
# Enable logging to file
|
||||||
|
set("log.file", log_to_file)
|
||||||
|
set("log.file.path", logpath)
|
||||||
|
|
||||||
|
# Set up emergency fallback track
|
||||||
|
emergency = single("/home/pi/RetroPie/mp3/Youtube/Dr. Peacock - Trip to Ireland [GvrvQTUbUcA].mp3")
|
||||||
|
radio = fallback(id="switcher2", track_sensitive=false, [s, emergency])
|
||||||
|
# Set up an interactive control for skipping tracks
|
||||||
|
p = interactive.bool("Skip track", false)
|
||||||
|
|
||||||
|
def http_skip(~protocol, ~data, ~headers, uri)=
|
||||||
|
radio.skip()
|
||||||
|
http.response(code=200,data="Skipped")
|
||||||
|
end
|
||||||
|
harbor.http.register(port=54321,method="GET","/skip", http_skip)
|
||||||
|
|
||||||
|
|
||||||
|
# Function to process the request queue and skip the current track if requested
|
||||||
|
def check_skip()
|
||||||
|
if p() then
|
||||||
|
log.info("Skipping current track.")
|
||||||
|
radio.skip()
|
||||||
|
p.set(false)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Run the queue processing function every 60 seconds
|
||||||
|
thread.run(every=60., queue_processing)
|
||||||
|
|
||||||
|
|
||||||
|
# Run the check_skip function every 5 seconds
|
||||||
|
thread.run(every=15., check_skip)
|
||||||
|
|
||||||
|
|
||||||
|
# Enable persistent script parameters
|
||||||
|
interactive.persistent("script.params")
|
||||||
|
|
||||||
|
# Configure output formats and destinations
|
||||||
|
|
||||||
|
output.icecast(%mp3, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="mp3-stream", radio)
|
||||||
|
output.pulseaudio(radio)
|
||||||
|
#output.file.hls("/tmp/hls", [("mp3-low", %mp3(bitrate=96)), ("mp3-hi", %mp3(bitrate=160))], radio)
|
||||||
|
# Uncomment the following lines to enable additional output formats
|
||||||
|
# output.icecast(%opus, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="opus-stream", radio)
|
||||||
|
# output.icecast(%ogg, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="ogg-stream", radio)
|
||||||
|
# output.icecast(%aac, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="aac-stream", radio)
|
||||||
|
# output.icecast(%flac, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="flac-stream", radio)
|
||||||
|
# output.icecast(%vorbis, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="vorbis-stream", radio)
|
||||||
|
# output.icecast(%speex, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="speex-stream", radio)
|
||||||
|
# output.icecast(%wav, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="wav-stream", radio)
|
||||||
|
# output.icecast(%pcm, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="pcm-stream", radio)
|
||||||
|
# output.icecast(%raw, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="raw-stream", radio)
|
||||||
|
# output.icecast(%s16l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s16l-stream", radio)
|
||||||
|
# output.icecast(%s16b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s16b-stream", radio)
|
||||||
|
# output.icecast(%s24l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s24l-stream", radio)
|
||||||
|
# output.icecast(%s24b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s24b-stream", radio)
|
||||||
|
# output.icecast(%s32l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s32l-stream", radio)
|
||||||
|
# output.icecast(%s32b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s32b-stream", radio)
|
||||||
|
# output.icecast(%s64l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s64l-stream", radio)
|
||||||
|
# output.icecast(%s64b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s64b-stream", radio)
|
||||||
|
# output.icecast(%s128l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s128l-stream", radio)
|
||||||
|
# output.icecast(%s128b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s128b-stream", radio)
|
||||||
|
# output.icecast(%s256l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s256l-stream", radio)
|
||||||
|
# output.icecast(%s256b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s256b-stream", radio)
|
||||||
|
# output.icecast(%s512l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s512l-stream", radio)
|
||||||
|
# output.icecast(%s512b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s512b-stream", radio)
|
||||||
|
# output.icecast(%s1024l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s1024l-stream", radio)
|
||||||
|
# output.icecast(%s1024b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s1024b-stream", radio)
|
||||||
|
# output.icecast(%s2048l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s2048l-stream", radio)
|
||||||
|
# output.icecast(%s2048b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s2048b-stream", radio)
|
||||||
|
# output.icecast(%s4096l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s4096l-stream", radio)
|
||||||
|
# output.icecast(%s4096b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s4096b-stream", radio)
|
||||||
|
# output.icecast(%s8192l, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s8192l-stream", radio)
|
||||||
|
# output.icecast(%s8192b, host="retropie", port=8000, password=credentials.password, icy_metadata="true", mount="s8192b-stream", radio)
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=radio_conjurer Liquidsoap daemon
|
||||||
|
After=network.target
|
||||||
|
Documentation=http://liquidsoap.fm/
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=pi
|
||||||
|
WorkingDirectory=/home/pi/Conjurer/
|
||||||
|
ExecStart=/home/pi/.opam/4.13.0/bin/liquidsoap /home/pi/Conjurer/radio_conjurer.liq
|
||||||
|
Restart=always
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
Alias=radio_conjurer-liquidsoap.service
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
flask
|
||||||
|
waitress
|
||||||
|
sshtunnel
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<title>HTTP Live Streaming Example</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<video src="/tmp/hls/stream.m3u8" height="300" width="400">
|
||||||
|
</video>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Executable
+6
@@ -0,0 +1,6 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd ./conjurer
|
||||||
|
git pull
|
||||||
|
cd ..
|
||||||
|
cp ./conjurer/conjurer_musician/conjurer_musician.py ./Conjurer
|
||||||
|
sudo systemctl restart conjurer_musician.service
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd ./conjurer
|
||||||
|
git pull
|
||||||
|
cd ..
|
||||||
|
cp ./conjurer/conjurer_musician/radio_conjurer.liq ./Conjurer
|
||||||
|
sudo systemctl restart radio_service.service
|
||||||
@@ -1,134 +0,0 @@
|
|||||||
"""
|
|
||||||
The provided Python script sets up a Flask web server to manage a list of music files, with
|
|
||||||
functions for rescanning the music folder, updating the music list, and serving the music list via
|
|
||||||
API endpoints.
|
|
||||||
"""
|
|
||||||
import json
|
|
||||||
from pathlib import Path
|
|
||||||
from platform import uname
|
|
||||||
import threading
|
|
||||||
import time
|
|
||||||
from sys import platform
|
|
||||||
import logging
|
|
||||||
from flask import Flask, jsonify, request
|
|
||||||
from waitress import serve
|
|
||||||
|
|
||||||
|
|
||||||
HOST_ADDRESS = "192.168.1.15"
|
|
||||||
PORT_ADDRESS = 5000
|
|
||||||
if platform in ("linux", "linux2"):
|
|
||||||
SEPARATOR_FILE_PATH = "/"
|
|
||||||
if "microsoft-standard" in uname().release:
|
|
||||||
LOGFILE = "/home/mtuszowski/conjurer/discord_mus_service.log"
|
|
||||||
MUSIC_FOLDER = "/mnt/g/Muzyka/"
|
|
||||||
NETRC_FILE = "/home/mtuszowski/.netrc"
|
|
||||||
LOGSTORE = "/home/mtuszowski/conjurer/logs/"
|
|
||||||
ENCODING = "utf-8"
|
|
||||||
GRAPHICS_PATH = "/home/mtuszowski/conjurer/Conjurer_graphics/"
|
|
||||||
|
|
||||||
else:
|
|
||||||
LOGFILE = "/home/pi/Conjurer/discord_mus_service.log"
|
|
||||||
MEMORY_FIVE_SIARA = "/home/pi/Conjurer/pamiec.json"
|
|
||||||
SYSTEM_GPT_SETTINGS = "/home/pi/Conjurer/system_gpt_settings.json"
|
|
||||||
MEMORY_FIVE_MUZYKA = "/home/pi/Conjurer/pamiec_muzyki.json"
|
|
||||||
MUSIC_FOLDER = "/home/pi/RetroPie/mp3/"
|
|
||||||
SETTINGS_FILE = "/home/pi/Conjurer/settings.json"
|
|
||||||
NETRC_FILE = "/home/pi/.netrc"
|
|
||||||
LOGSTORE = "/home/pi/RetroPie/logs/"
|
|
||||||
ACCIDENT_LOG = "/home/pi/Conjurer/accident_log.json"
|
|
||||||
ENCODING = "utf-8"
|
|
||||||
GRAPHICS_PATH = "/home/pi/RetroPie/Conjurer_graphics/"
|
|
||||||
DIR_PATH_SADOX = "/home/pi/RetroPie/Fansadox/"
|
|
||||||
|
|
||||||
|
|
||||||
music_file_list = []
|
|
||||||
def rescan():
|
|
||||||
"""
|
|
||||||
The `rescan` function logs a message, scans for mp3 files in a specified folder,
|
|
||||||
and adds them to a list of music files.
|
|
||||||
"""
|
|
||||||
logging.info("Rescan triggered")
|
|
||||||
for mp3_item in Path.glob(Path(MUSIC_FOLDER), "**/*.mp3"):
|
|
||||||
temp_music_file = mp3_item.as_posix()
|
|
||||||
if platform == "win32":
|
|
||||||
temp_music_file = temp_music_file.replace("/", "\\")
|
|
||||||
music_file_list.append(temp_music_file)
|
|
||||||
|
|
||||||
def thread_rescan():
|
|
||||||
"""
|
|
||||||
The `thread_rescan` function periodically triggers a rescan operation after a specified time
|
|
||||||
interval.
|
|
||||||
"""
|
|
||||||
logging.info("Starting filesystemupdater")
|
|
||||||
while True:
|
|
||||||
time.sleep(60*60*24)
|
|
||||||
logging.info("Rescan triggered")
|
|
||||||
rescan()
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
|
||||||
|
|
||||||
@app.route("/mp3", methods=["GET"])
|
|
||||||
def get_music_list():
|
|
||||||
"""
|
|
||||||
The function `get_music_list` returns a JSON object containing a list of music files.
|
|
||||||
:return: A JSON response containing a key "music_file_list" with the value of the variable
|
|
||||||
`music_file_list`.
|
|
||||||
"""
|
|
||||||
return jsonify({"music_file_list": music_file_list})
|
|
||||||
|
|
||||||
|
|
||||||
@app.route("/update_mp3", methods=["POST"])
|
|
||||||
def update_music_list():
|
|
||||||
"""
|
|
||||||
The function `update_music_list` receives a POST request with a JSON payload, logs the received
|
|
||||||
item, adds it to a music file list, and returns a success message along with the updated record.
|
|
||||||
:return: The function `update_music_list` is returning a tuple containing a JSON response and a
|
|
||||||
status code. The JSON response includes keys `isError`, `message`, `statusCode`, and `data`,
|
|
||||||
with values indicating the success of the operation and the data that was
|
|
||||||
received and added to the `music_file_list`.
|
|
||||||
The status code returned is 200, indicating a successful response.
|
|
||||||
"""
|
|
||||||
record = json.loads(request.data)
|
|
||||||
app.logger.info(record["item"])
|
|
||||||
music_file_list.append(record["item"])
|
|
||||||
return_data = (
|
|
||||||
jsonify(isError=False, message="Success", statusCode=200, data=record),
|
|
||||||
200,
|
|
||||||
)
|
|
||||||
return return_data
|
|
||||||
|
|
||||||
def flask_debug():
|
|
||||||
"""
|
|
||||||
The `flask_debug` function starts a Flask application in debug mode without using the reloader.
|
|
||||||
Do not use for production for fucks sake.
|
|
||||||
"""
|
|
||||||
# trunk-ignore(bandit/B201)
|
|
||||||
app.run(debug=True, use_reloader=False, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
|
||||||
|
|
||||||
def waitress_run():
|
|
||||||
"""
|
|
||||||
The `waitress_run` function serves the `app` on host "0.0.0.0"
|
|
||||||
and port 5000 using the Waitress WSGI server.
|
|
||||||
"""
|
|
||||||
serve(app, host=HOST_ADDRESS, port=PORT_ADDRESS)
|
|
||||||
|
|
||||||
def waitress_run():
|
|
||||||
serve(app, host="0.0.0.0", port=5000)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
logger = logging.getLogger('conjurer_musician')
|
|
||||||
logger.setLevel(logging.DEBUG)
|
|
||||||
rescan()
|
|
||||||
threads = []
|
|
||||||
threads.append(threading.Thread(target=waitress_run))
|
|
||||||
threads.append(threading.Thread(target=thread_rescan))
|
|
||||||
#threads.append(threading.Thread(target=flask_debug))
|
|
||||||
|
|
||||||
for worker in threads:
|
|
||||||
worker.start()
|
|
||||||
|
|
||||||
for worker in threads:
|
|
||||||
worker.join()
|
|
||||||
|
|
||||||
#TODO: wrzucić wyszukanie i robienie playlist
|
|
||||||
#TODO: Inne pliki - sadox, dokumenty naukowe itp.
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
[Unit]
|
|
||||||
Description=Conjurer_musician
|
|
||||||
After=multi-user.target
|
|
||||||
|
|
||||||
[Service]
|
|
||||||
Type=simple
|
|
||||||
ExecStart=/home/pi/Conjurer-service/conjurer/file_webservice/bin/python3 /home/pi/Conjurer-service/conjurer/file_webservice/conjurer_musician.py
|
|
||||||
Restart=on-abort
|
|
||||||
|
|
||||||
[Install]
|
|
||||||
WantedBy=multi-user.target
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
python3 -m venv /home/pi/Conjurer-service/conjurer/file_webservice
|
|
||||||
source ./bin/activate
|
|
||||||
./bin/python3 -m pip install --upgrade pip
|
|
||||||
./bin/python3 -m pip install -r requirements_musician.txt
|
|
||||||
deactivate
|
|
||||||
|
|
||||||
sudo cp ./conjurer_musician.service /etc/systemd/system/
|
|
||||||
sudo systemctl daemon-reload
|
|
||||||
sudo systemctl start conjurer_musician.service
|
|
||||||
sudo systemctl enable conjurer_musician.service
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
flask
|
|
||||||
waitress
|
|
||||||
+6
-2
@@ -1,5 +1,9 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
cd /home/pi
|
||||||
|
mdkir Conjurer
|
||||||
|
cd Conjurer
|
||||||
python3 -m venv /home/pi/Conjurer/env
|
python3 -m venv /home/pi/Conjurer/env
|
||||||
|
cp /home/pi/conjurer/requirements_bot.txt /home/pi/Conjurer/
|
||||||
source ./env/bin/activate
|
source ./env/bin/activate
|
||||||
./env/bin/python3 -m pip install --upgrade pip
|
./env/bin/python3 -m pip install --upgrade pip
|
||||||
./env/bin/python3 -m pip install -r requirements_bot.txt
|
./env/bin/python3 -m pip install -r requirements_bot.txt
|
||||||
@@ -8,5 +12,5 @@ sed -i '1i\import shutil' ./env/lib/python3.11/site-packages/spotify_dl/youtube.
|
|||||||
deactivate
|
deactivate
|
||||||
sudo cp ./conjurer.service /etc/systemd/system/
|
sudo cp ./conjurer.service /etc/systemd/system/
|
||||||
sudo systemctl daemon-reload
|
sudo systemctl daemon-reload
|
||||||
sudo systemctl start conjurer_musician.service
|
sudo systemctl start conjurer.service
|
||||||
sudo systemctl enable conjurer_musician.service
|
sudo systemctl enable conjurer.service
|
||||||
|
|||||||
Executable
+34
@@ -0,0 +1,34 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
cd /home/pi
|
||||||
|
mkdir Conjurer
|
||||||
|
cd Conjurer
|
||||||
|
python3 -m venv /home/pi/Conjurer/env
|
||||||
|
source /home/pi/Conjurer/env/bin/activate
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/requirements_musician.txt /home/pi/Conjurer/
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/conjurer_musician.py /home/pi/Conjurer/
|
||||||
|
cp /home/pi/conjurer/conjurer_musician/radio_conjurer.liq /home/pi/Conjurer/
|
||||||
|
touch /home/pi/Conjurer/discord_mus_service.log
|
||||||
|
|
||||||
|
touch /home/pi/Conjurer/playlist.json
|
||||||
|
touch /home/pi/Conjurer/prio_playlist.json
|
||||||
|
|
||||||
|
touch /home/pi/Conjurer/all_playlist.playlist
|
||||||
|
touch /home/pi/Conjurer/priority_queue.playlist
|
||||||
|
touch /home/pi/Conjurer/hit.playlist
|
||||||
|
touch /home/pi/Conjurer/request.playlist
|
||||||
|
touch /home/pi/Conjurer/persistence.log
|
||||||
|
echo "[]" > /home/pi/Conjurer/persistence.log
|
||||||
|
|
||||||
|
touch /home/pi/Conjurer/radio_log.log
|
||||||
|
./env/bin/python3 -m pip install --upgrade pip
|
||||||
|
./env/bin/python3 -m pip install -r requirements_musician.txt
|
||||||
|
deactivate
|
||||||
|
sudo cp /home/pi/conjurer/conjurer_musician/conjurer_musician.service /etc/systemd/system/
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl start conjurer_musician.service
|
||||||
|
sudo systemctl enable conjurer_musician.service
|
||||||
|
|
||||||
|
sudo cp /home/pi/conjurer/conjurer_musician/radio_service.service /etc/systemd/system/
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl start radio_service.service
|
||||||
|
sudo systemctl enable radio_service.service
|
||||||
@@ -10,4 +10,7 @@ PyPDF2
|
|||||||
requests
|
requests
|
||||||
spotipy
|
spotipy
|
||||||
tiktoken
|
tiktoken
|
||||||
PyNaCl
|
PyNaCl
|
||||||
|
flask
|
||||||
|
waitress
|
||||||
|
clickupython
|
||||||
Executable
+3
@@ -0,0 +1,3 @@
|
|||||||
|
sshpass -p raspberry ssh pi@192.168.1.15 << 'ENDSSH'
|
||||||
|
sudo reboot
|
||||||
|
ENDSSH
|
||||||
-2742
File diff suppressed because it is too large
Load Diff
@@ -204,6 +204,13 @@
|
|||||||
false,
|
false,
|
||||||
false
|
false
|
||||||
],
|
],
|
||||||
|
"same plusy\\b": [
|
||||||
|
"Jak na cmentarzu Szefie. Jak na cmentarzu",
|
||||||
|
15.0,
|
||||||
|
0,
|
||||||
|
false,
|
||||||
|
false
|
||||||
|
],
|
||||||
"kld\\b": [
|
"kld\\b": [
|
||||||
"KLD? Mieli tam burdel większy niż u Hammera. Ale o ile Hammer sprzątnął już nawet całkiem nieźle - to tam śmieci wyniosły się same. Uczcijmy ich minutą śmiechu.",
|
"KLD? Mieli tam burdel większy niż u Hammera. Ale o ile Hammer sprzątnął już nawet całkiem nieźle - to tam śmieci wyniosły się same. Uczcijmy ich minutą śmiechu.",
|
||||||
15.0,
|
15.0,
|
||||||
|
|||||||
@@ -239,6 +239,8 @@ def get_item_name(sp, item_type, item_id):
|
|||||||
name = sp.album(album_id=item_id).get("name")
|
name = sp.album(album_id=item_id).get("name")
|
||||||
elif item_type == "track":
|
elif item_type == "track":
|
||||||
name = sp.track(track_id=item_id).get("name")
|
name = sp.track(track_id=item_id).get("name")
|
||||||
|
elif item_type == "artist_top_tracks":
|
||||||
|
name = sp.artist_top_tracks(artist_id=item_id).get("name")
|
||||||
return sanitize(name)
|
return sanitize(name)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
+27
-9
@@ -15,10 +15,16 @@ the `smart_write` function returns the size of the encoded bytes of the input `l
|
|||||||
return any value.
|
return any value.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
FILE_SIZE_LIMIT = 100000 * 1024
|
FILE_SIZE_LIMIT = 50000 * 1024
|
||||||
SOURCE = "/mnt/n/scimag_2020-05-30.sql/scimag_2020-05-30.sql"
|
|
||||||
OUTPUT_FOLDER = "/mnt/n/scimag_2020-05-30.sql/"
|
|
||||||
OUTPUT_FILE_SCHEMA = "_scimag_chunk.sql"
|
SOURCE = r"C:\Database\dois-2022-02-12\sci-hub-doi-2022-02-12.txt"
|
||||||
|
OUTPUT_FOLDER = r"C:\Database\\chunks\\"
|
||||||
|
OUTPUT_FILE_SCHEMA = r"_chunk.txt"
|
||||||
|
|
||||||
|
#SOURCE = "/mnt/n/scimag_2020-05-30.sql/scimag_2020-05-30.sql"
|
||||||
|
#OUTPUT_FOLDER = "/mnt/n/scimag_2020-05-30.sql/"
|
||||||
|
#OUTPUT_FILE_SCHEMA = "_scimag_chunk.sql"
|
||||||
|
|
||||||
|
|
||||||
def smart_write(lines, file):
|
def smart_write(lines, file):
|
||||||
@@ -33,9 +39,12 @@ def smart_write(lines, file):
|
|||||||
:return: The function `smart_write` returns the size of the encoded bytes of the input `lines` in
|
:return: The function `smart_write` returns the size of the encoded bytes of the input `lines` in
|
||||||
bytes.
|
bytes.
|
||||||
"""
|
"""
|
||||||
encoded_bytes = lines.encode("utf-8")
|
try:
|
||||||
size_in_bytes = len(encoded_bytes)
|
encoded_bytes = lines.encode("utf-8")
|
||||||
file.write(lines)
|
size_in_bytes = len(encoded_bytes)
|
||||||
|
file.write(lines)
|
||||||
|
except:
|
||||||
|
return 0
|
||||||
|
|
||||||
return size_in_bytes
|
return size_in_bytes
|
||||||
|
|
||||||
@@ -59,8 +68,9 @@ def cut_my_life_into_pieces(source, output_folder, output_file):
|
|||||||
with open(source, encoding="utf-8") as infile:
|
with open(source, encoding="utf-8") as infile:
|
||||||
output_size = 0
|
output_size = 0
|
||||||
output_no = 0
|
output_no = 0
|
||||||
|
lineno = 0
|
||||||
for line in infile:
|
for line in infile:
|
||||||
|
lineno += 1
|
||||||
if output_size > FILE_SIZE_LIMIT:
|
if output_size > FILE_SIZE_LIMIT:
|
||||||
output_no += 1
|
output_no += 1
|
||||||
output_size = 0
|
output_size = 0
|
||||||
@@ -70,4 +80,12 @@ def cut_my_life_into_pieces(source, output_folder, output_file):
|
|||||||
"a",
|
"a",
|
||||||
encoding="utf-8",
|
encoding="utf-8",
|
||||||
) as op_file:
|
) as op_file:
|
||||||
output_size += smart_write(line, op_file)
|
tmp = smart_write(line, op_file)
|
||||||
|
if tmp > 0:
|
||||||
|
output_size += smart_write(line, op_file)
|
||||||
|
else:
|
||||||
|
print (f'Error at {lineno}')
|
||||||
|
print(line)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
cut_my_life_into_pieces(SOURCE,OUTPUT_FOLDER,OUTPUT_FILE_SCHEMA)
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
import json
|
||||||
|
MEMORY_FIVE_SIARA = "/home/pi/Conjurer/pamiec.json"
|
||||||
|
ENCODING = "utf-8"
|
||||||
|
|
||||||
|
messages = [
|
||||||
|
{"role": "user", "content" : "Hammer: All systems operational"},
|
||||||
|
{"role": "assistant", "content": "Aye, Szefie! Wszystkie systemy działają sprawnie. Gotów do dalszej służby! Co podać?"},
|
||||||
|
{"role": "user", "content" : "Hammer: Podaj coś na orzeźwienie i przebudzenie. COś co rozgrzeje moje stare kości po ciężkiej pracy!"},
|
||||||
|
{"role": "assistant", "content": "Z przyjemnością Szefie! Dla Ciebie przygotuję Midzynarodówkę Interstelarną Wolf's Dragoon - wysoce energetyczny koktajl rozgrzewający i pobudzający jednocześnie."+
|
||||||
|
"Do przygotowania potrzebuję:"+
|
||||||
|
"\n"+
|
||||||
|
"60 ml mocnego rumu z Wolfa (lub innego, jeśli wolisz)"+
|
||||||
|
"30 ml soku z cytryny"+
|
||||||
|
"35 ml domowego syropu z mięty"+
|
||||||
|
"15 ml świeżo zmielonej kawy"+
|
||||||
|
"\n"+
|
||||||
|
"Mieszam wszystko w shakerze z lodem, a potem przelewam przez sitko do kieliszka koktajlowego. W górę serwuję z bigoskiem z miętą i zestem z cytryny. Rozgrzewa, orzeźwia i daje kopa na resztę nocy."+
|
||||||
|
"\n"+
|
||||||
|
"Co powiesz Szefie, podać Ci taki?"},
|
||||||
|
{"role": "user", "content" : "Hammer: Brzmi jak coś co zrobi robotę! Skal!"},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
for temp in messages:
|
||||||
|
|
||||||
|
with open(MEMORY_FIVE_SIARA, "r+", encoding=ENCODING) as file_memory:
|
||||||
|
# First we load existing data into a dict.
|
||||||
|
file_data = json.load(file_memory)
|
||||||
|
# Join new_data with file_data inside emp_details
|
||||||
|
file_data.append(temp)
|
||||||
|
file_memory.seek(0)
|
||||||
|
# convert back to json.
|
||||||
|
json.dump(file_data, file_memory, indent=4)
|
||||||
Reference in New Issue
Block a user