reddit_client.cpython-310.pyc•4.93 kB
o
2f�h� � @ s> d Z ddlmZmZmZ ddlZddlmZ G dd� d�ZdS )zReddit API client wrapper.� )�Any�Dict�ListN� )� RedditConfigc @ s� e Zd ZdZdefdd�Zedd� �Z d d
ed ed e d
edede
e ee f f dd�Z
dede ee f fdd�Zd
ede ee f fdd�Zd d
ed e de
e ee f fdd�Z d d ed e d
edede
e ee f f
dd�ZdS ) � RedditClientz!Reddit API client for MCP server.�configc C s || _ d| _dS )z6Initialize Reddit client with read-only configuration.N)r �_reddit)�selfr � r �:/Users/lige/Desktop/reddit-mcp/reddit_mcp/reddit_client.py�__init__
s
zRedditClient.__init__c C s. | j du rtj| jj| jj| jjd�| _ | j S )z*Lazy initialize AsyncPRAW Reddit instance.N)� client_id�
client_secret�
user_agent)r � asyncprawZRedditr r r r )r
r r r �reddit s
�zRedditClient.reddit�
� relevance�all�subreddit_name�query�limit�sort� time_filter�returnc � s� �z_| j �|�I dH }g }|j||||d�}|2 zD3 dH W } | j| j| jr*t| j�nd| j| j| j d| j
� �| j | j t
| j�dkrJ| jdd� d n| j| j| jt| j�d�
}
|�|
� q6 |W S tyx } z td|� d t| �� ���d} ~ ww )
z Search for posts in a subreddit.N�r r r � [deleted]�https://reddit.com�� �...�
�id�title�author�score� upvote_ratio�url� permalink� created_utc� num_comments�selftext�is_self�domain� subredditzError searching posts in r/�: �r r. �searchr" r# r$ �strr% r& r' r( r) r* �lenr+ r, r- �append� Exception) r
r r r r r r. �posts�search_results�
submission� post_data�er r r � search_posts s@ � �
$� ���zRedditClient.search_posts�post_idc
� s� �zY| j j|d�I dH }i d|j�d|j�d|jr t|j�nd�d|j�d|j�d |j�d
d |j � ��d |j
�d
|j �d|j �d|j
�d|j�dt|j��d|j�d|j�d|j�W S tyr } z td|� dt|�� ���d}~ww )z/Get detailed information about a specific post.)r" Nr" r# r$ r r% r& r' r( r r) r* r+ r, r- r. Z
flair_text�locked�stickiedz Error getting post details for r/ )r r8 r" r# r$ r2 r% r& r' r( r) r* r+ r, r- r. Zlink_flair_textr= r> r5 )r
r< r8 r: r r r �get_post_detailsI sP ������� �� �
� � �
������z RedditClient.get_post_detailsc
� s� �z2| j �|�I dH }|j|jt|j�dkr |jdd� d n|j|j|j|j|j |j
d|j� �d� W S t yK } z t d|� dt |�� ���d}~ww )z"Get information about a subreddit.Nr r zhttps://reddit.com/r/) �namer# � description� subscribers�active_user_countr) �over18�public_descriptionr' z#Error getting subreddit info for r/r/ )
r r. � display_namer# r3 rA rB rC r) rD rE r5 r2 )r
r r. r: r r r �get_subreddit_infoh s" �$
� ��z RedditClient.get_subreddit_infoc � s� �zZ| j �|�I dH }g }|j|d�2 zD3 dH W }|j|j|jr%t|j�nd|j|j|j d|j
� �|j |j t
|j�dkrE|jdd� d n|j|j|jt|j�d�
}|�|� q6 |W S tys } z td|� d t|�� ���d}~ww )
z Get hot posts from a subreddit.N)r r r �� r r! z Error getting hot posts from r/r/ )r r. �hotr" r# r$ r2 r% r& r' r( r) r* r3 r+ r, r- r4 r5 )r
r r r. r6 r8 r9 r: r r r �
get_hot_posts| s4 �
$� ���zRedditClient.get_hot_postsc � s� �z_| j �d�I dH }g }|j||||d�}|2 zD3 dH W }|j|j|jr*t|j�nd|j|j|j d|j
� �|j |j t
|j�dkrJ|jdd� d n|j|j|jt|j�d�
} |�| � q6 |W S tyx }
z td |� d
t|
�� ���d}
~
ww ) z9Search for posts across all of Reddit (site-wide search).r Nr r r r r r! z&Error searching all Reddit for query 'z': r0 ) r
r r r r Z
all_subredditr6 r7 r8 r9 r: r r r �search_all_reddit� s@ ��
$� ���z RedditClient.search_all_redditN)r r r )r )�__name__�
__module__� __qualname__�__doc__r r
�propertyr r2 �intr r r r; r? rG rJ rK r r r r r
sL
������
�, $ ������r ) rO �typingr r r r r r r r r r r �<module> s