Update main.py

Import app variable from src/app.py

Author: CodeIter

main.py

@@ -1,324 +1,5 @@
-import typer
-import os
-import threading
-import json
-from rich.console import Console
-from rich.table import Table
-from datetime import datetime, timedelta, timezone
-from dotenv import load_dotenv
-
-from src.api import get_user_id, get_user_profile, get_user_posts, get_post_insights, fetch_all_posts, create_post, get_post_replies, get_post_replies_count
-from src.utils import convert_to_locale
-
-app = typer.Typer()
-console = Console()
-load_dotenv()
-
-ACCESS_TOKEN = os.getenv("ACCESS_TOKEN")
-HEADERS = {
-    'Authorization': f'Bearer {ACCESS_TOKEN}'
-}
-DRAFTS_FILE = 'drafts.json'
-SERVER_PROCESS_TIME = 10
-
-@app.command()
-def get_profile():
-    """
-    Retrieve and display user profile information, including the last post made by the user.
-    """
-    user_id = get_user_id(HEADERS)
-    profile = get_user_profile(user_id, HEADERS)
-    last_post = get_user_posts(user_id, HEADERS, limit=1)[0]
-
-    profile_table = Table(title=f'{profile["username"]}\'s Profile')
-    profile_table.add_column("Field", style="cyan", no_wrap=True)
-    profile_table.add_column("Value", style="magenta")
-
-    profile_table.add_row("ID", profile.get("id", "N/A"))
-    profile_table.add_row("Username", profile.get("username", "N/A"))
-    profile_table.add_row("Profile Picture URL", profile.get("threads_profile_picture_url", "N/A"))
-    profile_table.add_row("Biography", profile.get("threads_biography", "N/A"))
-    if last_post:
-        profile_table.add_row("Last Post ID", last_post.get("id", "N/A"))
-        profile_table.add_row("Post Type", last_post.get("media_type", "N/A"))
-        profile_table.add_row("Post Text", last_post.get("text", "N/A"))
-        profile_table.add_row("Post Permalink", last_post.get("permalink", "N/A"))
-        profile_table.add_row("Post Timestamp", convert_to_locale(last_post.get("timestamp", "N/A")))
-    else:
-        profile_table.add_row("Message", "No posts found")
-
-    console.print(profile_table)
-
-@app.command()
-def get_recent_posts(limit: int = 5):
-    """
-    Retrieve the most recent posts.
-    """
-    user_id = get_user_id(HEADERS)
-    posts = get_user_posts(user_id, HEADERS, limit=limit)
-
-    table = Table(title="Recent Posts")
-    table.add_column("ID", style="cyan", no_wrap=True)
-    table.add_column("Username", style="cyan", no_wrap=True)
-    table.add_column("Timestamp", style="magenta")
-    table.add_column("Type", style="green")
-    table.add_column("Text", style="yellow")
-    table.add_column("Permalink", style="blue")
-    table.add_column("Replies", style="red")
-
-    for post in posts:
-        if post.get('media_type') == 'REPOST_FACADE':
-            continue
-        timestamp = convert_to_locale(post.get('timestamp', 'N/A'))
-        replies_count = get_post_replies_count(post['id'], HEADERS)
-        table.add_row(
-            post.get('id', 'N/A'),
-            post.get('username', 'N/A'),
-            timestamp,
-            post.get('media_type', 'N/A'),
-            post.get('text', 'N/A'),
-            post.get('permalink', 'N/A'),
-            str(replies_count)
-        )
-
-    console.print(table)
-
-@app.command()
-def get_top_liked_posts(limit: int = 5, time_range: str = None):
-    """
-    Retrieve the top liked posts of all time or within a specific time range.
-    """
-    user_id = get_user_id(HEADERS)
-    all_posts = fetch_all_posts(user_id, HEADERS)
-
-    if time_range:
-        now = datetime.now(timezone.utc)
-        if time_range.endswith('w'):
-            weeks = int(time_range[:-1])
-            start_time = now - timedelta(weeks=weeks)
-        elif time_range.endswith('d'):
-            days = int(time_range[:-1])
-            start_time = now - timedelta(days=days)
-        elif time_range.endswith('h'):
-            hours = int(time_range[:-1])
-            start_time = now - timedelta(hours=hours)
-        elif time_range.endswith('m'):
-            months = int(time_range[:-1])
-            start_time = now - timedelta(days=30 * months)
-        else:
-            typer.echo("Invalid time range format. Use '2w' for 2 weeks, '7d' for 7 days, '24h' for 24 hours, or '7m' for 7 months.")
-            return
-
-        all_posts = [post for post in all_posts if datetime.strptime(post['timestamp'], '%Y-%m-%dT%H:%M:%S%z') >= start_time]
-
-    posts_with_likes = []
-    for post in all_posts:
-        if post.get('media_type') == 'REPOST_FACADE':
-            continue
-        insights = get_post_insights(post['id'], HEADERS)
-        if 'likes' in insights:
-            posts_with_likes.append((post, insights['likes']))
-
-    posts_with_likes.sort(key=lambda x: x[1], reverse=True)
-    top_liked_posts = posts_with_likes[:limit]
-
-    table = Table(title="Top Liked Posts")
-    table.add_column("Username", style="cyan", no_wrap=True)
-    table.add_column("Timestamp", style="magenta")
-    table.add_column("Type", style="green")
-    table.add_column("Text", style="yellow")
-    table.add_column("Permalink", style="blue")
-    table.add_column("Likes", style="red")
-    table.add_column("Replies", style="green")
-    table.add_column("Reposts", style="blue")
-    table.add_column("Quotes", style="yellow")
-    table.add_column("Views", style="cyan")
-
-    for post, likes in top_liked_posts:
-        timestamp = convert_to_locale(post.get('timestamp', 'N/A'))
-        insights = get_post_insights(post['id'], HEADERS)
-        table.add_row(
-            post.get('username', 'N/A'),
-            timestamp,
-            post.get('media_type', 'N/A'),
-            post.get('text', 'N/A'),
-            post.get('permalink', 'N/A'),
-            str(insights.get('likes', 'N/A')),
-            str(insights.get('replies', 'N/A')),
-            str(insights.get('reposts', 'N/A')),
-            str(insights.get('quotes', 'N/A')),
-            str(insights.get('views', 'N/A'))
-        )
-
-    console.print(table)
-
-@app.command()
-def create_text_post(text: str):
-    """
-    Create a post with text.
-    """
-    user_id = get_user_id(HEADERS)
-    payload = {
-        "media_type": "TEXT",
-        "text": text
-    }
-    post = create_post(user_id, HEADERS, payload)
-    typer.echo(f"Post created with ID: {post['id']}")
-
-@app.command()
-def create_image_post(text: str, image_url: str):
-    """
-    Create a post with an image.
-    """
-    user_id = get_user_id(HEADERS)
-    payload = {
-        "media_type": "IMAGE",
-        "image_url": image_url,
-        "text": text
-    }
-    post = create_post(user_id, HEADERS, payload)
-    typer.echo(f"Post created with ID: {post['id']}")
-
-@app.command()
-def get_latest_replies(media_id: str, limit: int = 5):
-    """
-    Retrieve the latest replies for a specific media post.
-    """
-    replies = get_post_replies(media_id, HEADERS, limit=limit)
-
-    table = Table(title="Latest Replies")
-    table.add_column("Username", style="cyan", no_wrap=True)
-    table.add_column("Media ID", style="cyan", no_wrap=True)
-    table.add_column("Timestamp", style="magenta")
-    table.add_column("Text", style="yellow")
-    table.add_column("Permalink", style="blue")
-
-    for reply in replies:
-        timestamp = convert_to_locale(reply.get('timestamp', 'N/A'))
-        table.add_row(
-            reply.get('username', 'N/A'),
-            reply.get('id', 'N/A'),
-            timestamp,
-            reply.get('text', 'N/A'),
-            reply.get('permalink', 'N/A')
-        )
-
-    console.print(table)
-
-@app.command()
-def send_reply(media_id: str, text: str):
-    """
-    Send a reply to a specific media post.
-    """
-    user_id = get_user_id(HEADERS)
-    payload = {
-        "media_type": "TEXT",
-        "text": text,
-        "reply_to_id": media_id
-    }
-    reply = create_post(user_id, HEADERS, payload)
-    typer.echo(f"Reply created with ID: {reply['id']}")
-
-def job_create_text_post(text: str):
-    """
-    Job function to create a post with text.
-    """
-    create_text_post(text)
-
-@app.command()
-def schedule_post(text: str, post_time: str):
-    """
-    Schedule a post with text at a specific time.
-    """
-    post_time_dt = datetime.strptime(post_time, '%Y-%m-%d %H:%M:%S')
-    current_time = datetime.now()
-    delay = (post_time_dt - current_time).total_seconds()
-
-    if delay <= 0:
-        typer.echo("Scheduled time must be in the future.")
-        return
-
-    timer = threading.Timer(delay, job_create_text_post, [text])
-    timer.start()
-    typer.echo(f"Post scheduled for {post_time} with text: '{text}'")
-
-@app.command()
-def create_draft(text: str, drafts_file: str = DRAFTS_FILE):
-    '''
-    Create a draft with the given text and save it to the drafts file.
-    '''
-    if os.path.exists(drafts_file):
-        with open(drafts_file, 'r') as file:
-            drafts = json.load(file)
-    else:
-        drafts = []
-
-    next_id = max([draft['id'] for draft in drafts], default=0) + 1
-
-    draft = {
-        "id": next_id,
-        "text": text,
-        "timestamp": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-    }
-    drafts.append(draft)
-
-    with open(drafts_file, 'w') as file:
-        json.dump(drafts, file, indent=4)
-
-    typer.echo(f"Draft created with ID: {next_id}")
-
-@app.command()
-def get_drafts(drafts_file: str = DRAFTS_FILE):
-    '''
-    Get all drafts from the drafts file.
-    '''
-    if not os.path.exists(drafts_file):
-        typer.echo("No drafts found.")
-        return
-
-    with open(drafts_file, 'r') as file:
-        drafts = json.load(file)
-
-    table = Table(title="Drafts")
-    table.add_column("ID", style="cyan", no_wrap=True)
-    table.add_column("Text", style="yellow")
-    table.add_column("Timestamp", style="magenta")
-
-    for draft in drafts:
-        table.add_row(
-            str(draft['id']),
-            draft['text'],
-            draft['timestamp']
-        )
-
-    console.print(table)
-
-@app.command()
-def send_draft(draft_id: int, drafts_file: str = DRAFTS_FILE):
-    '''
-    Send a draft with the given ID and remove it from the drafts file.
-    '''
-    if not os.path.exists(drafts_file):
-        typer.echo("No drafts found.")
-        raise typer.Exit(1)
-
-    with open(drafts_file, 'r') as file:
-        drafts = json.load(file)
-
-    draft = next((draft for draft in drafts if draft['id'] == draft_id), None)
-
-    if draft is None:
-        typer.echo(f"Draft with ID {draft_id} not found.")
-        raise typer.Exit(1)
-
-    create_text_post(draft['text'])
-
-    drafts = [draft for draft in drafts if draft['id'] != draft_id]
-
-    with open(drafts_file, 'w') as file:
-        json.dump(drafts, file, indent=4)
-
-    typer.echo(f"Draft with ID {draft_id} sent and removed from drafts.")
+import src.env
+from src.app import app
 
 def main():
     app()