message.py

"""create a message for a tweet"""

import contextlib
import random

import pyshorteners
import requests

from constants import (BAR_CHART, CHART_DECREASING, CHART_INCREASING,
                       DOWN_ARROW, END_MESSAGE, FIRE, GITHUB_LINK, GLOBE,
                       HASHTAGS, NUMBERS, PAGE_FACE_UP, PERSON,
                       REPLY_MESSAGE_FOCUS, REPLY_MESSAGE_PEAK,
                       REPLY_MESSAGE_TOP, REPLY_MESSAGE_TRENDING, RIGHT_ARROW,
                       SHOPPING_CART, TROPHY, VIDEO_GAME)


def create_message_top_games(games):
    """Create a message for a tweet."""
    message = f'Top 10 most played #games on #Steam currently {TROPHY}{GLOBE}\n\n'
    games_names = []
    for item in games.items():
        rank = get_rank(item[1]['rank'])
        games_names.append(item[0])
        current_players = item[1]['current_players']
        message += ((f'{rank}-{item[0]} (Currently : {current_players} {PERSON})\n'))
        # add a hashtag for each game (without spaces and special characters)
        message += '#' + ''.join(e for e in item[0] if e.isalnum()) + '\n'
    return add_hashtags(message, games_names)


def create_message_peak_of_the_day(games):
    """Create a message for a tweet."""
    message = f'Peak of players today on #Steam !{GLOBE}{TROPHY}\n\n'
    games_names = []
    sorted_games = sorted(
        games.items(), key=lambda x: x[1]['peak_players'], reverse=True)
    sorted_games = dict(sorted_games)
    for i, item in enumerate(sorted_games.items()):
        rank = get_rank(i+1)
        games_names.append(item[0])
        peak_players = item[1]['peak_players']
        message += (
            (f'{rank} {item[0]} (Peak of players: {peak_players} {PERSON}{FIRE})\n'))
        # add a hashtag for each game (without spaces and special characters)
        message += '#' + ''.join(e for e in item[0] if e.isalnum()) + '\n'
    return add_hashtags(message, games_names)


def create_message_trending_games(games):
    """Create a message for a tweet."""
    message = f'Top 5 trending #games on #Steam currently {TROPHY}\n\n'
    games_names = []
    for item in games.items():
        rank = get_rank(item[1]['rank'])
        evolution = item[1]['evolution']
        games_names.append(item[0])
        message += (
            (f'{rank} {item[0]} (Last 24 hours : {evolution} {CHART_INCREASING})\n'))
        # add a hashtag for each game (without spaces and special characters)
        message += '#' + ''.join(e for e in item[0] if e.isalnum()) + '\n'
    return add_hashtags(message, games_names)


def create_message_game_focus(game):
    """Create a message for a tweet."""
    message = f'Focus on {game["name"]} over the last month {VIDEO_GAME}{BAR_CHART}\n'
    message += '#' + \
        ''.join(e for e in game['name'] if e.isalnum()) + ' #Steam\n\n'
    message += f'Peak of players : {game["peak_players"]} {PERSON}{FIRE}\n'
    message += f'The average number of players : {int(float(game["avg_players"]))}'\
        f'{PERSON}{GLOBE}\n'
    if game['gain'].startswith('+'):
        message += f'Evolution gain compared with the last month : {game["gain"]} '\
            f'{CHART_INCREASING}\n'
    else:
        message += f'Evolution loss compared with the last month : {game["gain"]} '\
            f'{CHART_DECREASING}\n'
    if game['description'] != 'No description available.':
        message += f'Description {PAGE_FACE_UP} : \n'
        for text in game['description']:
            message += f'{text} '
            if text.endswith('.') or text.endswith('!') or text.endswith('?'):
                message += '\n'
    else:
        message += f'{game["description"]} for {game["name"]}\n'
    message += '\n\n'
    message += f'Link : {game["steam_link"]} {SHOPPING_CART}\n'
    message += create_reply_message([game['name']], "focus")[0]
    return message


def create_reply_message(games, tweet_type):
    """Create a reply message for a tweet."""
    if tweet_type == "trending":
        return add_hashtags(REPLY_MESSAGE_TRENDING, games)
    if tweet_type == "peak":
        return add_hashtags(REPLY_MESSAGE_PEAK, games)
    if tweet_type == "focus":
        return add_hashtags(REPLY_MESSAGE_FOCUS, games)
    return add_hashtags(REPLY_MESSAGE_TOP, games)


def create_links_message(links, games):
    """Create a message to display links of games"""
    message = f'Links of the #games in the #leaderboard #today '\
        f'{TROPHY}{VIDEO_GAME}\n\n'
    for i, link in enumerate(links):
        with contextlib.suppress(requests.exceptions.ReadTimeout):
            link = pyshorteners.Shortener().tinyurl.short(link)
        line = f'{games[i]} {RIGHT_ARROW} {link} {SHOPPING_CART}\n'
        message += line
    message += '\n'
    message += END_MESSAGE
    return message


def add_hashtags(message, games):
    """add hashtags to the message"""
    message += '\n'
    hashtag = ''
    for i in range(1, 7):
        while hashtag in message:
            hashtag = random.choice(HASHTAGS)
        message += f'{hashtag} '
        if i % 2 == 0:
            message += '\n'
    message += '\n'
    # return the main message and the games names
    return message, games


def get_rank(rank):
    """getting the rank of the game in emojis"""
    if rank == 1:
        rank = "\U0001F947"
    elif rank == 2:
        rank = "\U0001F948"
    elif rank == 3:
        rank = "\U0001F949"
    else:
        rank = NUMBERS[rank]
    return rank


def cut_message(message):
    """cut the message into many messages with 280 characters"""
    messages = message.split('\n')
    final_messages = []
    while messages != []:
        message = ''
        if len(messages[0]) > 250:
            messages[0] = f'{messages[0][:250]}...'
        # cut messages into 280 characters messages
        while len(message) + len(messages[0]) < 240:
            message += messages[0] + '\n'
            messages.pop(0)
            if messages == []:
                break
        with contextlib.suppress(IndexError):
            if message[-2] != '\n':
                message += '\n'
        message += GITHUB_LINK + ' ' + DOWN_ARROW
        final_messages.append(message)
    # final messages contains messages cut into 280 characters
    return final_messages