Refactor and clean up

Author: ptrlrd

README.md

@@ -1,125 +1,70 @@
-
-# Threads CLI
-
-This CLI tool allows you to interact with the Threads API to perform various actions like creating posts, 
-getting profile information, retrieving recent posts, scheduling posts, and managing drafts. 
-The tool is built using Python and leverages the `typer` library for command-line interface creation and the `rich` 
-library for pretty-printing tables.
-
-## Installation
-
-1. **Clone the repository:**
-    ```bash
-    git clone https://github.com/your-repo/threads-cli.git
-    cd threads-cli
-    ```
-
-2. **Set up a virtual environment:**
-    ```bash
-    python3 -m venv .venv
-    source .venv/bin/activate
-    ```
-
-3. **Install the dependencies:**
-    ```bash
-    pip install -r requirements.txt
-    ```
-
-4. **Set up environment variables:**
-    Create a `.env` file in the root directory with the following content:
-    ```env
-    ACCESS_TOKEN=your_access_token
-    ```
+# Threads Command-Line Interface (CLI) Tool
+The Threads CLI Tool is a powerful and efficient command-line application designed to provide another way to interact
+with Threads. Developed using Python and leveraging the `typer` library for a user-friendly interface and the `rich`
+library for visual output, this tool can do a range of actions, including creating posts, retrieving 
+profile information, managing drafts, and more.
+
+## Key Features
+- User Profile: Retrieve and display comprehensive user profile information, including the user's last post.
+- Recent Posts: Fetch and showcase the most recent posts made by the user.
+- Top Liked Posts: Retrieve the top liked posts of all time or within a specific time range.
+- Post Creation: Create text posts and image posts.
+- Replies: Get the latest replies for a specific media post and send replies to engage with the community.
+- Scheduling: Schedule posts to be automatically published at a specified future date and time.
+- Drafts: Create, manage, and send drafts, providing flexibility in your content creation workflow.
+
+## Getting Started
+To get started with the Threads CLI Tool, follow these steps:
+
+Clone the repository:
+```
+git clone https://github.com/your-repo/threads-cli.git
+cd threads-cli
+```
+
+Set up a virtual environment:
+```
+python3 -m venv .venv
+source .venv/bin/activate
+```
+
+Install the required dependencies:
+```
+pip install -r requirements.txt
+```
+
+Configure environment variables:
+Create a `.env` file in the project's root directory and add the following:
+
+
+```
+ACCESS_TOKEN=your_access_token
+```
 
 ## Usage
-
-### Commands
-If any of these example commands are incorrect, if you do `python main.py --help` you will see the list of commands and their descriptions.
-
-
-1. **Get Profile Information:**
-    Retrieve and display user profile information, including the last post made by the user.
-    ```bash
-    python main.py get-profile
-    ```
-
-2. **Get Recent Posts:**
-    Retrieve the most recent posts.
-    ```bash
-    python main.py get-recent-posts --limit 5
-    ```
-
-3. **Get Top Liked Posts:**
-    Retrieve the top liked posts of all time or within a specific time range.
-    ```bash
-    python main.py get-top-liked-posts --limit 5 --time-range 2w
-    ```
-
-4. **Create a Text Post:**
-    Create a post with text.
-    ```bash
-    python main.py create-text-post "This is a text post."
-    ```
-
-5. **Create an Image Post:**
-    Create a post with an image.
-    ```bash
-    python main.py create-image_post "This is an image post." "https://example.com/image.jpg"
-    ```
-
-6. **Get Latest Replies:**
-    Retrieve the latest replies for a specific media post.
-    ```bash
-    python main.py get-latest-replies --media-id <media_id> --limit 5
-    ```
-
-7. **Send a Reply:**
-    Send a reply to a specific media post.
-    ```bash
-    python main.py send-reply --media-id <media_id> --text "This is a reply."
-    ```
-
-8. **Schedule a Post:**
-    Schedule a post with text at a specific time.
-    ```bash
-    python main.py schedule-post --text "This is a scheduled post." --post-time "2024-06-22 23:22:00"
-    ```
-
-9. **Create a Draft:**
-    Create a draft with text.
-    ```bash
-    python main.py create-draft "This is a draft."
-    ```
-
-10. **Get Drafts:**
-    Retrieve all drafts.
-    ```bash
-    python main.py get-drafts
-    ```
-
-11. **Send a Draft:**
-    Send a draft with a specific ID and remove it from drafts.
-    ```bash
-    python main.py send-draft --draft-id 1
-    ```
-
-## Additional Information
-
-### Dependencies
-- `typer`
-- `requests`
-- `rich`
-- `python-dotenv`
-
-### Environment Variables
-- `ACCESS_TOKEN`: Your Threads API access token.
-
-## Development
-
-Feel free to contribute to this project by creating pull requests. Make sure to follow the standard coding guidelines and thoroughly test your changes.
-
-### Things to do
-- [ ] Add tests for the CLI commands.
-- [ ] Add support for creating polls.
-- [ ] Validate that creating videos/image posts works
-
+The Threads CLI Tool provides a wide range of commands to interact with the Threads API. Here are a few examples:
+
+Retrieve user profile:
+```
+python main.py get-profile
+```
+
+Get recent posts:
+```
+python main.py get-recent-posts --limit 10
+```
+
+Create a text post:
+```
+python main.py create-text-post "Hello, Threads!"
+```
+
+Schedule a post:
+```
+python main.py schedule-post "Scheduled post" "2024-06-22 09:00:00"
+```
+
+For a complete list of available commands and their descriptions, run:
+```
+python main.py --help
+```

api.py

@@ -0,0 +1,66 @@
+import requests
+from utils import convert_to_locale
+
+BASE_URL = "https://graph.threads.net/v1.0"
+
+def get_user_id(headers):
+    response = requests.get(f"{BASE_URL}/me?fields=id", headers=headers)
+    response.raise_for_status()
+    return response.json()['id']
+
+def get_user_profile(user_id, headers):
+    profile_fields = "id,username,threads_profile_picture_url,threads_biography"
+    response = requests.get(f"{BASE_URL}/{user_id}?fields={profile_fields}", headers=headers)
+    response.raise_for_status()
+    return response.json()
+
+def get_user_posts(user_id, headers, limit=5):
+    post_fields = "id,media_product_type,media_type,media_url,permalink,owner,username,text,timestamp,shortcode,thumbnail_url,children,is_quote_post"
+    response = requests.get(f"{BASE_URL}/{user_id}/threads?fields={post_fields}&limit={limit}", headers=headers)
+    response.raise_for_status()
+    return response.json().get('data', [])
+
+def get_post_insights(post_id, headers):
+    metrics = "views,likes,replies,reposts,quotes"
+    response = requests.get(f"{BASE_URL}/{post_id}/insights?metric={metrics}", headers=headers)
+    if response.status_code == 500:
+        return {}
+    response.raise_for_status()
+    insights = response.json().get('data', [])
+    return {insight['name']: insight['values'][0]['value'] for insight in insights}
+
+def fetch_all_posts(user_id, headers):
+    all_posts = []
+    fields = "id,media_product_type,media_type,media_url,permalink,owner,username,text,timestamp,shortcode,thumbnail_url,children,is_quote_post"
+    next_url = f"{BASE_URL}/{user_id}/threads?fields={fields}"
+    while next_url:
+        response = requests.get(next_url, headers=headers)
+        response.raise_for_status()
+        data = response.json()
+        all_posts.extend(data.get('data', []))
+        next_url = data.get('paging', {}).get('next')
+    return all_posts
+
+def create_post(user_id, headers, payload):
+    response = requests.post(f"{BASE_URL}/{user_id}/threads", headers=headers, json=payload)
+    response.raise_for_status()
+    container_id = response.json()['id']
+
+    publish_payload = {
+        "creation_id": container_id
+    }
+    publish_response = requests.post(f"{BASE_URL}/{user_id}/threads_publish", headers=headers, json=publish_payload)
+    publish_response.raise_for_status()
+    return publish_response.json()
+
+def get_post_replies(media_id, headers, limit=5):
+    fields = "id,text,username,permalink,timestamp,media_product_type,media_type,shortcode,has_replies,root_post,replied_to,is_reply,hide_status"
+    response = requests.get(f"{BASE_URL}/{media_id}/replies?fields={fields}&limit={limit}&reverse=true", headers=headers)
+    response.raise_for_status()
+    return response.json().get('data', [])
+
+def get_post_replies_count(post_id, headers):
+    response = requests.get(f"{BASE_URL}/{post_id}/replies", headers=headers)
+    response.raise_for_status()
+    replies = response.json().get('data', [])
+    return len(replies)