mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-07-08 20:19:57 +02:00
added inactive following dvm
This commit is contained in:
@ -28,6 +28,7 @@ TASK_IMAGE_GENERATION_NIP89_DTAG2 = "fdgdfg"
|
|||||||
TASK_IMAGE_GENERATION_NIP89_DTAG3 = "asdasd"
|
TASK_IMAGE_GENERATION_NIP89_DTAG3 = "asdasd"
|
||||||
TASK_SPEECH_TO_TEXT_NIP89 = "asdasdas"
|
TASK_SPEECH_TO_TEXT_NIP89 = "asdasdas"
|
||||||
TASK_MEDIA_CONVERTER_NIP89_DTAG = "asdasdasd"
|
TASK_MEDIA_CONVERTER_NIP89_DTAG = "asdasdasd"
|
||||||
|
TASK_DISCOVER_INACTIVE_NIP89_DTAG = "sdfdfsdf12312"
|
||||||
|
|
||||||
|
|
||||||
#Backend Specific Options for tasks that require inputs, such as Endpoints or API Keys
|
#Backend Specific Options for tasks that require inputs, such as Endpoints or API Keys
|
||||||
|
@ -138,7 +138,8 @@ class Bot:
|
|||||||
|
|
||||||
# send the event to the DVM
|
# send the event to the DVM
|
||||||
send_event(nip90request, client=self.client, dvm_config=self.dvm_config)
|
send_event(nip90request, client=self.client, dvm_config=self.dvm_config)
|
||||||
print(nip90request.as_json())
|
#print(nip90request.as_json())
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
elif decrypted_text.lower().startswith("balance"):
|
elif decrypted_text.lower().startswith("balance"):
|
||||||
|
7
main.py
7
main.py
@ -11,7 +11,7 @@ from nostr_sdk import Keys
|
|||||||
from bot.bot import Bot
|
from bot.bot import Bot
|
||||||
from playground import build_pdf_extractor, build_googletranslator, build_unstable_diffusion, build_sketcher, \
|
from playground import build_pdf_extractor, build_googletranslator, build_unstable_diffusion, build_sketcher, \
|
||||||
build_dalle, \
|
build_dalle, \
|
||||||
build_whisperx, build_libretranslator, build_external_dvm, build_media_converter
|
build_whisperx, build_libretranslator, build_external_dvm, build_media_converter, build_inactive_follows_finder
|
||||||
from utils.definitions import EventDefinitions
|
from utils.definitions import EventDefinitions
|
||||||
from utils.dvmconfig import DVMConfig
|
from utils.dvmconfig import DVMConfig
|
||||||
|
|
||||||
@ -99,6 +99,11 @@ def run_nostr_dvm_with_local_config():
|
|||||||
bot_config.SUPPORTED_DVMS.append(media_bringer) # We also add Sketcher to the bot
|
bot_config.SUPPORTED_DVMS.append(media_bringer) # We also add Sketcher to the bot
|
||||||
media_bringer.run()
|
media_bringer.run()
|
||||||
|
|
||||||
|
#Spawn DVM10 Discover inactive followers
|
||||||
|
discover_inactive = build_inactive_follows_finder("Bygones")
|
||||||
|
bot_config.SUPPORTED_DVMS.append(discover_inactive) # We also add Sketcher to the bot
|
||||||
|
discover_inactive.run()
|
||||||
|
|
||||||
|
|
||||||
Bot(bot_config)
|
Bot(bot_config)
|
||||||
|
|
||||||
|
@ -5,10 +5,11 @@ from nostr_sdk import PublicKey, Keys
|
|||||||
|
|
||||||
from interfaces.dvmtaskinterface import DVMTaskInterface
|
from interfaces.dvmtaskinterface import DVMTaskInterface
|
||||||
from tasks.convert_media import MediaConverter
|
from tasks.convert_media import MediaConverter
|
||||||
|
from tasks.discovery_inactive_follows import DiscoverInactiveFollows
|
||||||
from tasks.imagegeneration_openai_dalle import ImageGenerationDALLE
|
from tasks.imagegeneration_openai_dalle import ImageGenerationDALLE
|
||||||
from tasks.imagegeneration_sdxl import ImageGenerationSDXL
|
from tasks.imagegeneration_sdxl import ImageGenerationSDXL
|
||||||
from tasks.textextraction_whisperx import SpeechToTextWhisperX
|
from tasks.textextraction_whisperx import SpeechToTextWhisperX
|
||||||
from tasks.textextractionpdf import TextExtractionPDF
|
from tasks.textextraction_pdf import TextExtractionPDF
|
||||||
from tasks.translation_google import TranslationGoogle
|
from tasks.translation_google import TranslationGoogle
|
||||||
from tasks.translation_libretranslate import TranslationLibre
|
from tasks.translation_libretranslate import TranslationLibre
|
||||||
from utils.admin_utils import AdminConfig
|
from utils.admin_utils import AdminConfig
|
||||||
@ -161,7 +162,7 @@ def build_unstable_diffusion(name):
|
|||||||
nip89config = NIP89Config()
|
nip89config = NIP89Config()
|
||||||
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG")
|
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG")
|
||||||
nip89config.CONTENT = json.dumps(nip89info)
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
||||||
admin_config=admin_config, options=options)
|
admin_config=admin_config, options=options)
|
||||||
|
|
||||||
|
|
||||||
@ -195,7 +196,7 @@ def build_whisperx(name):
|
|||||||
nip89config = NIP89Config()
|
nip89config = NIP89Config()
|
||||||
nip89config.DTAG = os.getenv("TASK_SPEECH_TO_TEXT_NIP89")
|
nip89config.DTAG = os.getenv("TASK_SPEECH_TO_TEXT_NIP89")
|
||||||
nip89config.CONTENT = json.dumps(nip89info)
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
return SpeechToTextWhisperX(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
return SpeechToTextWhisperX(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
||||||
admin_config=admin_config, options=options)
|
admin_config=admin_config, options=options)
|
||||||
|
|
||||||
|
|
||||||
@ -230,7 +231,7 @@ def build_sketcher(name):
|
|||||||
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG2")
|
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG2")
|
||||||
nip89config.CONTENT = json.dumps(nip89info)
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
|
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
|
||||||
return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
||||||
admin_config=admin_config, options=options)
|
admin_config=admin_config, options=options)
|
||||||
|
|
||||||
|
|
||||||
@ -262,7 +263,8 @@ def build_dalle(name):
|
|||||||
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG3")
|
nip89config.DTAG = os.getenv("TASK_IMAGE_GENERATION_NIP89_DTAG3")
|
||||||
nip89config.CONTENT = json.dumps(nip89info)
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
|
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
|
||||||
return ImageGenerationDALLE(name=name, dvm_config=dvm_config, nip89config=nip89config, admin_config=admin_config)
|
return ImageGenerationDALLE(name=name, dvm_config=dvm_config, nip89config=nip89config, admin_config=admin_config)
|
||||||
|
|
||||||
|
|
||||||
def build_media_converter(name):
|
def build_media_converter(name):
|
||||||
dvm_config = DVMConfig()
|
dvm_config = DVMConfig()
|
||||||
@ -284,14 +286,51 @@ def build_media_converter(name):
|
|||||||
}
|
}
|
||||||
|
|
||||||
nip89config = NIP89Config()
|
nip89config = NIP89Config()
|
||||||
new_dtag = nip89_create_d_tag(name, Keys.from_sk_str(dvm_config.PRIVATE_KEY).public_key().to_hex(), nip89info["image"])
|
new_dtag = nip89_create_d_tag(name, Keys.from_sk_str(dvm_config.PRIVATE_KEY).public_key().to_hex(),
|
||||||
|
nip89info["image"])
|
||||||
print("Some new dtag:" + new_dtag)
|
print("Some new dtag:" + new_dtag)
|
||||||
nip89config.DTAG = os.getenv("TASK_MEDIA_CONVERTER_NIP89_DTAG")
|
nip89config.DTAG = os.getenv("TASK_MEDIA_CONVERTER_NIP89_DTAG")
|
||||||
nip89config.CONTENT = json.dumps(nip89info)
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
return MediaConverter(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
return MediaConverter(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
||||||
admin_config=admin_config)
|
admin_config=admin_config)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def build_inactive_follows_finder(name):
|
||||||
|
dvm_config = DVMConfig()
|
||||||
|
dvm_config.PRIVATE_KEY = os.getenv("NOSTR_PRIVATE_KEY7")
|
||||||
|
dvm_config.LNBITS_INVOICE_KEY = os.getenv("LNBITS_INVOICE_KEY")
|
||||||
|
dvm_config.LNBITS_URL = os.getenv("LNBITS_HOST")
|
||||||
|
# Add NIP89
|
||||||
|
nip90params = {
|
||||||
|
"user": {
|
||||||
|
"required": False,
|
||||||
|
"values": [],
|
||||||
|
"description": "Do the task for another user"
|
||||||
|
},
|
||||||
|
"since_days": {
|
||||||
|
"required": False,
|
||||||
|
"values": [],
|
||||||
|
"description": "The number of days a user has not been active to be considered inactive"
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nip89info = {
|
||||||
|
"name": name,
|
||||||
|
"image": "https://image.nostr.build/c33ca6fc4cc038ca4adb46fdfdfda34951656f87ee364ef59095bae1495ce669.jpg",
|
||||||
|
"about": "I discover users you follow, but that have been inactive on Nostr",
|
||||||
|
"nip90Params": nip90params
|
||||||
|
}
|
||||||
|
|
||||||
|
nip89config = NIP89Config()
|
||||||
|
new_dtag = nip89_create_d_tag(name, Keys.from_sk_str(dvm_config.PRIVATE_KEY).public_key().to_hex(),
|
||||||
|
nip89info["image"])
|
||||||
|
print("Some new dtag:" + new_dtag)
|
||||||
|
nip89config.DTAG = os.getenv("TASK_DISCOVER_INACTIVE_NIP89_DTAG")
|
||||||
|
nip89config.CONTENT = json.dumps(nip89info)
|
||||||
|
return DiscoverInactiveFollows(name=name, dvm_config=dvm_config, nip89config=nip89config,
|
||||||
|
admin_config=admin_config)
|
||||||
|
|
||||||
def build_external_dvm(name, pubkey, task, kind, fix_cost, per_unit_cost):
|
def build_external_dvm(name, pubkey, task, kind, fix_cost, per_unit_cost):
|
||||||
dvm_config = DVMConfig()
|
dvm_config = DVMConfig()
|
||||||
dvm_config.PUBLIC_KEY = PublicKey.from_hex(pubkey).to_hex()
|
dvm_config.PUBLIC_KEY = PublicKey.from_hex(pubkey).to_hex()
|
||||||
|
170
tasks/discovery_inactive_follows.py
Normal file
170
tasks/discovery_inactive_follows.py
Normal file
@ -0,0 +1,170 @@
|
|||||||
|
import json
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
from datetime import timedelta
|
||||||
|
from threading import Thread
|
||||||
|
|
||||||
|
from nostr_sdk import Client, Timestamp, PublicKey, Tag, Keys, Options, Alphabet
|
||||||
|
|
||||||
|
from interfaces.dvmtaskinterface import DVMTaskInterface
|
||||||
|
from utils.admin_utils import AdminConfig
|
||||||
|
from utils.definitions import EventDefinitions
|
||||||
|
from utils.dvmconfig import DVMConfig
|
||||||
|
from utils.nip89_utils import NIP89Config
|
||||||
|
from utils.nostr_utils import get_event_by_id
|
||||||
|
|
||||||
|
"""
|
||||||
|
This File contains a Module to find inactive follows for a user on nostr
|
||||||
|
|
||||||
|
Accepted Inputs: None needed
|
||||||
|
Outputs: A list of users that have been inactive
|
||||||
|
Params: None
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class DiscoverInactiveFollows(DVMTaskInterface):
|
||||||
|
KIND: int = EventDefinitions.KIND_NIP90_PEOPLE_DISCOVERY
|
||||||
|
TASK: str = "inactive-follows"
|
||||||
|
FIX_COST: float = 50
|
||||||
|
client: Client
|
||||||
|
dvm_config: DVMConfig
|
||||||
|
|
||||||
|
def __init__(self, name, dvm_config: DVMConfig, nip89config: NIP89Config,
|
||||||
|
admin_config: AdminConfig = None, options=None):
|
||||||
|
super().__init__(name, dvm_config, nip89config, admin_config, options)
|
||||||
|
|
||||||
|
def is_input_supported(self, tags):
|
||||||
|
# no input required
|
||||||
|
return True
|
||||||
|
|
||||||
|
def create_request_form_from_nostr_event(self, event, client=None, dvm_config=None):
|
||||||
|
self.client = client
|
||||||
|
self.dvm_config = dvm_config
|
||||||
|
|
||||||
|
request_form = {"jobID": event.id().to_hex()}
|
||||||
|
|
||||||
|
# default values
|
||||||
|
user = event.pubkey().to_hex()
|
||||||
|
since_days = 90
|
||||||
|
|
||||||
|
for tag in event.tags():
|
||||||
|
if tag.as_vec()[0] == 'param':
|
||||||
|
param = tag.as_vec()[1]
|
||||||
|
if param == "user": # check for param type
|
||||||
|
user = tag.as_vec()[2]
|
||||||
|
elif param == "since_days": # check for param type
|
||||||
|
since_days = int(tag.as_vec()[2])
|
||||||
|
|
||||||
|
options = {
|
||||||
|
"user": user,
|
||||||
|
"since_days": since_days
|
||||||
|
}
|
||||||
|
request_form['options'] = json.dumps(options)
|
||||||
|
return request_form
|
||||||
|
|
||||||
|
def process(self, request_form):
|
||||||
|
from nostr_sdk import Filter
|
||||||
|
from types import SimpleNamespace
|
||||||
|
ns = SimpleNamespace()
|
||||||
|
|
||||||
|
options = DVMTaskInterface.set_options(request_form)
|
||||||
|
step = 20
|
||||||
|
|
||||||
|
followers_filter = Filter().author(PublicKey.from_hex(options["user"])).kind(3).limit(1)
|
||||||
|
followers = self.client.get_events_of([followers_filter], timedelta(seconds=self.dvm_config.RELAY_TIMEOUT))
|
||||||
|
|
||||||
|
if len(followers) > 0:
|
||||||
|
result_list = []
|
||||||
|
newest = 0
|
||||||
|
best_entry = followers[0]
|
||||||
|
for entry in followers:
|
||||||
|
if entry.created_at().as_secs() > newest:
|
||||||
|
newest = entry.created_at().as_secs()
|
||||||
|
best_entry = entry
|
||||||
|
|
||||||
|
print(best_entry.as_json())
|
||||||
|
followings = []
|
||||||
|
ns.dic = {}
|
||||||
|
for tag in best_entry.tags():
|
||||||
|
if tag.as_vec()[0] == "p":
|
||||||
|
following = tag.as_vec()[1]
|
||||||
|
followings.append(following)
|
||||||
|
ns.dic[following] = "False"
|
||||||
|
print("Followings: " + str(len(followings)))
|
||||||
|
|
||||||
|
not_active_since_seconds = int(options["since_days"]) * 24 * 60 * 60
|
||||||
|
dif = Timestamp.now().as_secs() - not_active_since_seconds
|
||||||
|
not_active_since = Timestamp.from_secs(dif)
|
||||||
|
|
||||||
|
def scanList(users, instance, i, st, notactivesince):
|
||||||
|
from nostr_sdk import Filter
|
||||||
|
|
||||||
|
keys = Keys.from_sk_str(self.dvm_config.PRIVATE_KEY)
|
||||||
|
opts = Options().wait_for_send(True).send_timeout(
|
||||||
|
timedelta(seconds=10)).skip_disconnected_relays(False)
|
||||||
|
cli = Client.with_opts(keys, opts)
|
||||||
|
for relay in self.dvm_config.RELAY_LIST:
|
||||||
|
cli.add_relay(relay)
|
||||||
|
cli.connect()
|
||||||
|
|
||||||
|
filters = []
|
||||||
|
for i in range(i + st):
|
||||||
|
filter1 = Filter().author(PublicKey.from_hex(users[i])).since(notactivesince).limit(1)
|
||||||
|
filters.append(filter1)
|
||||||
|
event_from_authors = cli.get_events_of(filters, timedelta(seconds=10))
|
||||||
|
for author in event_from_authors:
|
||||||
|
instance.dic[author.pubkey().to_hex()] = "True"
|
||||||
|
print(str(i) + "/" + str(len(users)))
|
||||||
|
cli.disconnect()
|
||||||
|
|
||||||
|
threads = []
|
||||||
|
begin = 0
|
||||||
|
# Spawn some threads to speed things up
|
||||||
|
while begin < len(followings) - step:
|
||||||
|
args = [followings, ns, begin, step, not_active_since]
|
||||||
|
t = Thread(target=scanList, args=args)
|
||||||
|
threads.append(t)
|
||||||
|
begin = begin + step
|
||||||
|
|
||||||
|
# last to step size
|
||||||
|
missing_scans = (len(followings) - begin)
|
||||||
|
args = [followings, ns, begin, missing_scans, not_active_since]
|
||||||
|
t = Thread(target=scanList, args=args)
|
||||||
|
threads.append(t)
|
||||||
|
|
||||||
|
# Start all threads
|
||||||
|
for x in threads:
|
||||||
|
x.start()
|
||||||
|
|
||||||
|
# Wait for all of them to finish
|
||||||
|
for x in threads:
|
||||||
|
x.join()
|
||||||
|
|
||||||
|
result = {k for (k, v) in ns.dic.items() if v == "False"}
|
||||||
|
|
||||||
|
if len(result) == 0:
|
||||||
|
print("Not found")
|
||||||
|
return "No inactive followers found on relays."
|
||||||
|
print("Inactive accounts found: " + str(len(result)))
|
||||||
|
for k in result:
|
||||||
|
p_tag = Tag.parse(["p", k])
|
||||||
|
result_list.append(p_tag.as_vec())
|
||||||
|
|
||||||
|
return json.dumps(result_list)
|
||||||
|
|
||||||
|
def post_process(self, result, event):
|
||||||
|
"""Overwrite the interface function to return a social client readable format, if requested"""
|
||||||
|
for tag in event.tags():
|
||||||
|
if tag.as_vec()[0] == 'output':
|
||||||
|
format = tag.as_vec()[1]
|
||||||
|
if format == "text/plain": # check for output type
|
||||||
|
result_list = json.loads(result)
|
||||||
|
inactive_follows_list = ""
|
||||||
|
for tag in result_list:
|
||||||
|
p_tag = Tag.parse(tag)
|
||||||
|
inactive_follows_list = inactive_follows_list + "nostr:" + PublicKey.from_hex(
|
||||||
|
p_tag.as_vec()[1]).to_bech32() + "\n"
|
||||||
|
return inactive_follows_list
|
||||||
|
|
||||||
|
# if not text/plain, don't post-process
|
||||||
|
return result
|
@ -11,7 +11,7 @@ from utils.nostr_utils import get_event_by_id
|
|||||||
def input_data_file_duration(event, dvm_config, client, start=0, end=0):
|
def input_data_file_duration(event, dvm_config, client, start=0, end=0):
|
||||||
#print("[" + dvm_config.NIP89.NAME + "] Getting Duration of the Media file..")
|
#print("[" + dvm_config.NIP89.NAME + "] Getting Duration of the Media file..")
|
||||||
input_value = ""
|
input_value = ""
|
||||||
input_type = "url"
|
input_type = ""
|
||||||
for tag in event.tags():
|
for tag in event.tags():
|
||||||
if tag.as_vec()[0] == 'i':
|
if tag.as_vec()[0] == 'i':
|
||||||
input_value = tag.as_vec()[1]
|
input_value = tag.as_vec()[1]
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
from datetime import timedelta
|
from datetime import timedelta
|
||||||
|
from hashlib import sha256
|
||||||
|
|
||||||
from nostr_sdk import Tag, Keys, EventBuilder, Filter, Alphabet, PublicKey, Event
|
from nostr_sdk import Tag, Keys, EventBuilder, Filter, Alphabet, PublicKey, Event
|
||||||
|
|
||||||
@ -15,10 +16,13 @@ class NIP89Config:
|
|||||||
|
|
||||||
|
|
||||||
def nip89_create_d_tag(name, pubkey, image):
|
def nip89_create_d_tag(name, pubkey, image):
|
||||||
import hashlib
|
#import hashlib
|
||||||
m = hashlib.md5()
|
#m = hashlib.md5()
|
||||||
m.update(str(name + image + pubkey).encode("utf-8"))
|
#m.update(str(name + image + pubkey).encode("utf-8"))
|
||||||
d_tag = m.hexdigest()[0:16]
|
#d_tag = m.hexdigest()[0:16]
|
||||||
|
|
||||||
|
key_str = str(name + image + pubkey)
|
||||||
|
d_tag = sha256(key_str.encode('utf-8')).hexdigest()[:16]
|
||||||
return d_tag
|
return d_tag
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user