htmloutput.py

import aiofiles
import datetime
import math
import os
import re

import styledprint
import winelist


_indentcount = 0
_INDENTSTEP = 4


def writeline(line, nl=False):
    newline = ''
    for i in range(0, _INDENTSTEP * _indentcount):
        newline += ' '
    newline += line + os.linesep
    if (nl):
        newline += os.linesep
    return newline


def increase_indent_count():
    global _indentcount
    _indentcount += 1


def decrease_indent_count():
    global _indentcount
    _indentcount -= 1


def get_date():
    data = writeline('function getDate(){')
    increase_indent_count()
    data += writeline('return "{}";'.format(datetime.date.today().isoformat()))
    decrease_indent_count()
    data += writeline('};', True)
    return data


def get_rows(games):
    reviewMapping     = dict()
    reviewMapping[10] = 'Overwhelmingly Positive'
    reviewMapping[9]  = 'Very Positive'
    reviewMapping[8]  = 'Positive'
    reviewMapping[7]  = 'Mostly Positive'
    reviewMapping[6]  = 'Mixed'
    reviewMapping[5]  = 'Mostly Negative'
    reviewMapping[4]  = 'Negative'
    reviewMapping[3]  = 'Very Negative'
    reviewMapping[2]  = 'Overwhelmingly Negative'
    justifyFormat     = '<div class=\\"text\\">{0}</div>'

    data = writeline('function getRows(){')
    increase_indent_count()
    data += writeline('rows=[];', True)


    for gameName in sorted(games, key=str.lower):
        game = games[gameName]
        if (not game.hfr.is_available):
            continue

        data += writeline('var row = {')
        increase_indent_count()
        data += writeline('name: "{0}",'.format(gameName.replace('"', '\\"')))
        #TODO make smallImage an input parameter?
        smallImage = True
        if (game.store.link):
            if(game.store.image):
                if (smallImage):
                    imageLink = re.sub(r'header\.jpg.*', 'capsule_sm_120.jpg', game.store.image)
                else:
                    imageLink = game.store.image
                data += writeline('nameFormat: "<a href=\\"{0}\\"><b>{1}</b><img src=\\"{2}\\" width=\\"100%\\"/></a>",'
                                  .format(game.store.link, justifyFormat, imageLink))
            else:
                data += writeline('nameFormat: "<a href=\\"{0}\\"><b>{1}</b></a>",'
                                  .format(game.store.link, justifyFormat))
        else:
            data += writeline('nameFormat: "<b>{0}</b>",'.format(justifyFormat))
        if(game.store.description):
            data += writeline('description: "{0}",'.format(
                game.store.description.replace(os.linesep, '<br/>').replace('"', '\\"')))
        if (game.store.category):
            data += writeline('category: "{0}",'.format(game.store.category.name))
        if (len(game.store.os) > 0):
            data += writeline('os: "{0}",'.format(', '.join(game.store.os).replace('OS X', 'OS')))
        if (game.store.price != None):
            data += writeline('price: {0},'.format(str(game.store.price)))
            if (game.store.price == 0):
                data += writeline('priceFormat: "{0}",'
                                  .format(justifyFormat.replace('{0}', 'Free')))
            elif (game.store.price < 0):
                data += writeline('priceFormat: "{0}",'
                                  .format(justifyFormat.replace('{0}', 'N/A')))
            else:
                data += writeline('priceFormat: "{0}",'.
                                  format(justifyFormat.replace('{0}', '${0}')))

        if (len(game.store.genres) > 0):
            data += writeline('genres: "{0}",'.format(', '.join(game.store.genres)))
        if (game.store.release_date):
            date, errors = game.store.release_date
            # if errors has more than whitespace or empty strings
            errors = tuple(filter(lambda x: x.strip(), errors))
            if (len(errors)):
                fmt = '%Y'
            else:
                fmt = '%Y-%m-%d'
            data += writeline('date: "{0}",'.format(date.strftime(fmt).replace('-', '&#8209;')))
        if (game.store.avg_review) and (game.store.avg_review in reviewMapping):
            avg_review_text = reviewMapping[game.store.avg_review]
            MIN_POW         = 6
            avg_review_p    = math.pow(10, MIN_POW) * game.store.avg_review
            # if we have average negative reviews, we reduce the average review's power per review
            if (game.store.avg_review < 6):
                avg_review = str(avg_review_p - game.store.cnt_review)
            # if we have average positive reviews, we increment the average review's power per review
            else:
                avg_review = str(avg_review_p + game.store.cnt_review)
            if (game.store.avg_review < 10):
                avg_review = '0{0}'.format(avg_review)
            data += writeline('review: "{0} {1}",'.
                              format(avg_review, avg_review_text))
            data += writeline('reviewFormat: "{0}",'.
                              format(justifyFormat.replace('{0}',
                                                           '{0} ({1})'.
                                                           format(avg_review_text,
                                                                  game.store.cnt_review))))
        else:
            if (game.store.avg_review):
                styledprint.print_info('The average review {0} for game {1} is not in the mapping!'.
                      format(game.store.avg_review, gameName))

        if (game.hfr.requirements):
            if (game.hfr.requirements.lower() == 'standard'):
                stars = '*'
            else:
                stars = '**'

            if (game.hfr.is_new):
                requirements = '{0}: Nouveauté'.format(game.hfr.requirements)
            else:
                requirements = game.hfr.requirements
            data += writeline('requirements: "{0}<sup><b>{1}</b></sup>",'.format(requirements, stars))

        if (len(game.store.tags) > 0):
            data += writeline('tags: "{0}",'.format(', '.join(game.store.tags)))

        if (len(game.store.details) > 0):
            data += writeline('details: "{0}",'.format(', '.join(game.store.details)))

        if (game.hfr.gift_date):
            data += writeline('giftdate: "{0}",'.format(game.hfr.gift_date.strftime("%Y-%m-%d")))

        if (len(game.store.interface) > 0):
            data += writeline('interface: "{0}",'.format(', '.join(game.store.interface)))
        if (len(game.store.audio) > 0):
            data += writeline('audio: "{0}",'.format(', '.join(game.store.audio)))
        if (len(game.store.subtitles) > 0):
            data += writeline('subtitles: "{0}",'.format(', '.join(game.store.subtitles)))

        if (game.wine):
            data += writeline('wine: "{0}",'.format(', '
                .join([app.rating for app in game.wine])))
            data += writeline('wineFormat: "{0}",'
                              .format(justifyFormat
                                  .format(', '.join(
                                      ['<a href=\\"{0}\\"><b>{1}</b></a>'
                                           .format(app.link, app.rating) for app in game.wine]))))
        else:
            data += writeline('wineFormat: "{0}",'
                              .format(justifyFormat
                                      .format('<a href=\\"https://appdb.winehq.org/objectManager.php?sClass=application&sTitle=Browse+Applications&iappFamily-appNameOp=2&sappFamily-appNameData={0}\\"><i>?</i></a>'
                                              .format(gameName.replace('"', '\\"')))))

        decrease_indent_count()
        data += writeline('};')
        try:
            if (game.store.override):
                data += writeline('row[\'row-cls\'] = \'override\'')
        except:
            pass
        data += writeline('rows.push(row);', True)

    data += writeline('return rows;')
    decrease_indent_count()
    data += writeline('};')
    return data


async def output_to_js(dryrun, games, path):

    text = get_date() + os.linesep + get_rows(games)

    # These 2 chars proc a unicode encode error on Windows so we replace them
    text = text.replace('\x97', '')
    text = text.replace('\u2032', '\'')

    if (not dryrun):
        async with aiofiles.open(path, 'w', encoding='utf8') as f:
            await f.write(text.strip())