Skip to content

Commit

Permalink
Added recommender component
Browse files Browse the repository at this point in the history
  • Loading branch information
qstokkink committed Oct 7, 2024
1 parent 7d4a25e commit 22751a2
Show file tree
Hide file tree
Showing 16 changed files with 1,040 additions and 2 deletions.
40 changes: 40 additions & 0 deletions src/tribler/core/components.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,6 +282,46 @@ def finalize(self, ipv8: IPv8, session: Session, community: Community) -> None:
session.rest_manager.get_endpoint("/api/ipv8").endpoints["/dht"].dht = community


@precondition('session.config.get("recommender/enabled")')
@overlay("tribler.core.recommender.community", "RecommenderCommunity")
class RecommenderComponent(CommunityLauncherWEndpoints):
"""
Launch instructions for the user recommender community.
"""

def get_kwargs(self, session: Session) -> dict:
"""
Create and forward the rendezvous database for the Community.
"""
from tribler.core.recommender.manager import Manager

db_path = str(Path(session.config.get_version_state_dir()) / "sqlite" / "recommender.db")
if session.config.get("memory_db"):
db_path = ":memory:"

out = super().get_kwargs(session)
out["manager"] = Manager(db_path)

return out

def finalize(self, ipv8: IPv8, session: Session, community: Community) -> None:
"""
When we are done launching, register our REST API.
"""
from tribler.core.recommender.community import RecommenderCommunity

endpoint = session.rest_manager.get_endpoint("/api/recommender")
endpoint.manager = cast(RecommenderCommunity, community).manager

def get_endpoints(self) -> list[RESTEndpoint]:
"""
Add the knowledge endpoint.
"""
from tribler.core.recommender.restapi.endpoint import RecommenderEndpoint

return [*super().get_endpoints(), RecommenderEndpoint()]


@set_in_session("tunnel_community")
@precondition('session.config.get("tunnel_community/enabled")')
@after("DHTDiscoveryComponent")
Expand Down
Empty file.
Loading

0 comments on commit 22751a2

Please sign in to comment.