server.cpython-310.pyc•7.67 kB
o
Cl�h�' � @ sd U d Z ddlZddlZddlZddlmZmZmZmZ ddl m
Z
ddl m Z ddl
mZ ejejd� e�e�Ze
d �Zda
ee ed
< d d � Ze��
d$dedededededef dd��Ze��
d$dededededef
dd��Ze�� dedefdd ��Z e�� dedefd d ��Z e�� d%dededefd d ��Z d!d"� Z ed#kr�e � dS dS )&z/Reddit MCP Server implementation using FastMCP.� N)�Any�Dict�List�Optional)�FastMCP� )� RedditConfig)� RedditClient)�levelzreddit-mcp-tool�
reddit_clientc
C s` zt �� } t| �at�d� W dS ty/ } zt�dt|�� �� daW Y d}~dS d}~ww )z0Initialize the Reddit client with configuration.z8Reddit client initialized successfully in read-only modez$Failed to initialize Reddit client: N) r Zfrom_envr r �logger�info� Exception�error�str)�config�e� r �3/Users/lige/Desktop/reddit-mcp/reddit_mcp/server.py�initialize_reddit_client s ��r �
� relevance�all� subreddit�query�limit�sort� time_filter�returnc � s� �t du rdS z�t j| |t|d�||d�I dH }|s$d| � d|� d�W S dt|�� d | � d|� d
�}t|d �D ]]\}}||� d |d
� d|d � d|d � d|d d�d|d � d|d � d|d � d�7 }|d r�t|d �d kr�t|d �d kr�|d dd � d n|d } |d | � d�7 }|d7 }q7|W S ty� }
z t�d!| � d"t|
�� �� d!| � d"t|
�� �W Y d}
~
S d}
~
ww )#a�
Search for posts in a specific subreddit
Args:
subreddit: The name of the subreddit to search in (without r/)
query: The search query
limit: Number of posts to return (default: 10, max: 100)
sort: Sort method - "relevance", "hot", "top", "new", "comments" (default: "relevance")
time_filter: Time filter - "all", "day", "week", "month", "year" (default: "all")
Returns:
Human readable string containing search results
N��Error: Reddit client not initialized. Please check your configuration and ensure REDDIT_CLIENT_ID, REDDIT_CLIENT_SECRET, and REDDIT_USER_AGENT are set.�d )Zsubreddit_namer r r r zNo posts found in r/z
for query: '�'�Found z posts in r/�'
r �. **�title�**
Author: �author�
Score: �score� (upvote ratio: � upvote_ratio�.0%�)
Comments: � num_comments�
Link: � permalink�
Subreddit: r/r �
�selftextr �� �...� Content: zError searching posts in r/�: ) r Z search_posts�min�len� enumerater r r r ) r r r r r �posts�result�i�post�previewr r r r �search_reddit_posts$ sP � � ������� ,
��r@ c
� st �t du rdS z�t j| t|d�||d�I dH }|s d| � d�W S dt|�� d| � d �}t|d
�D ]]\}}||� d |d � d
|d � d|d � d|d d�d|d � d|d � d|d � d�7 }|d r�t|d �d kr�t|d �d kr}|d dd � d n|d }|d |� d�7 }|d7 }q0|W S ty� } z t�d | � d!t| �� �� d | � d!t| �� �W Y d} ~ S d} ~ ww )"a�
Search for posts across all of Reddit (site-wide search)
Args:
query: The search query to search across all Reddit
limit: Number of posts to return (default: 10, max: 100)
sort: Sort method - "relevance", "hot", "top", "new", "comments" (default: "relevance")
time_filter: Time filter - "all", "day", "week", "month", "year" (default: "all")
Returns:
Human readable string containing search results from across Reddit
Nr r )r r r r z)No posts found across Reddit for query: 'r! r" z% posts across all Reddit for query: 'r# r r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r r2 r3 r r4 r5 r6 z&Error searching all Reddit for query 'z': ) r Zsearch_all_redditr8 r9 r: r r r r )
r r r r r; r< r= r> r? r r r r �search_reddit_alla sN � �������� ,
��rA �post_idc � s �t du rdS z^t �| �I dH }d|d � d|d � d|d � d |d
d �d |d
� d|d � d|d � d|d � d|d � d|d � d�}|�d�rU|d|d � d�7 }|d rc|d |d � d�7 }|W S ty� } z t�d | � d t|�� �� d | � d t|�� �W Y d}~S d}~ww ) z�
Get detailed information about a specific Reddit post
Args:
post_id: The Reddit post ID
Returns:
Human readable string containing detailed post information
Nr z**r% z **
Author: r' z
Score: r) r* r+ r, z )
Comments: r. z
Link: r0 z
Subreddit: r/r z
Domain: Zdomainz
Locked: �lockedz
Stickied: Zstickiedr2 Z
flair_textzFlair: r3 z
Content:
z Error getting post details for r7 )r Zget_post_details�getr r r r )rB Z post_detailsr< r r r r �get_reddit_post_details� sF � ����������
��rE c � s� �t du rdS zJt �| �I dH }d|d � d|d � d|d d �d
|d p&d � d
|d � d|d � d|d � d�}|d rO|d |d krO|d|d � d�7 }|W S tyz } z t�d| � dt|�� �� d| � dt|�� �W Y d}~S d}~ww )z�
Get information about a subreddit
Args:
subreddit: The name of the subreddit (without r/)
Returns:
Human readable string containing subreddit information
Nr z**r/�namez **
Title: r% z
Subscribers: Z subscribers�,z
Active Users: Zactive_user_countzN/Az
NSFW: Zover18z
URL: Zurlz
Description:
Zpublic_descriptionr2 � descriptionz
Full Description:
z#Error getting subreddit info for r/r7 )r �get_subreddit_infor r r r )r Zsubreddit_infor< r r r r rI � s6 � ��
�����
��rI c � sX �t du rdS z{t �| t|d��I dH }|s d| � �W S d| � d�}t|d�D ]X\}}||� d|d � d
|d � d |d
� d|d d�d|d � d|d � d�7 }|d r{t|d �dkr{t|d �dkro|d dd� d n|d }|d|� d�7 }|d7 }q'|W S ty� } z t�d| � d t|�� �� d| � d t|�� �W Y d}~S d}~ww ) z�
Get hot posts from a subreddit
Args:
subreddit: The name of the subreddit (without r/)
limit: Number of posts to return (default: 10, max: 100)
Returns:
Human readable string containing hot posts
Nr r zNo hot posts found in r/zHot posts from r/z:
r r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r2 r3 r � r5 r6 z Error getting hot posts from r/r7 ) r Z
get_hot_postsr8 r: r9 r r r r )r r r; r< r= r> r? r r r r �get_hot_reddit_posts� s@ � ������,
��rK c C s t � t�� dS )z Entry point for the CLI command.N)r �mcp�runr r r r �
run_server s rN �__main__)r r r )r ) �__doc__Zjson�logging�os�typingr r r r Zfastmcpr r r r r � basicConfig�INFO� getLogger�__name__r rL �__annotations__r Ztoolr �intr@ rA rE rI rK rN r r r r �<module> sn
������ �<����� �9)$,
�