writeTextBox.py

from pygame import *
import pprint

import covid
import WebScrape
import destination



#global font variable
font.init()
title_font = font.SysFont('Calibri', 80)
body_font = font.SysFont('Calibri', 50)

testbodysurface = body_font.render("Test", False, (0,0,0))

'''Images'''
icon_covid = image.load("Images/CovidIcon.png")
icon_country = image.load("Images/CountryIcon.png")
icon_tourist = image.load("Images/TourtistIcon.png")

#resize images to needs
icon_covid = transform.scale(icon_covid, (testbodysurface.get_height(), testbodysurface.get_height()))
icon_country = transform.scale(icon_country, (testbodysurface.get_height(), testbodysurface.get_height()))
icon_tourist = transform.scale(icon_tourist, (testbodysurface.get_height(), testbodysurface.get_height()))


#thanks stackoverflow for this code (pygame suks)
'''
params:
- surface to blit on
- string
- (x,y) tuple
- font object
- color optional
blits text and accounts for line wrapping on the surface
returns the y value of the lowest text that is reached
'''




def blit_text(surface, text, pos, font, color=Color('black')):
    words = [word.split(' ') for word in text.splitlines()]  # 2D array where each row is a list of words.
    space = font.size(' ')[0]  # The width of a space.
    max_width, max_height = surface.get_size()
    x, y = pos
    for line in words:
        for word in line:
            word_surface = font.render(word, 0, color)
            word_width, word_height = word_surface.get_size()
            if x + word_width >= max_width:
                x = pos[0]  # Reset the x.
                y += word_height  # Start on new row.
            surface.blit(word_surface, (x, y))
            x += word_width + space
        x = pos[0]  # Reset the x.
        y += word_height  # Start on new row.

    return y  #bottom of lowest text written

        
'''
params:
country -> string with name
width, height -> dimensions of text box
returns a transparent surface with the text information centered on it
later will need to be blitted to the surface
'''
def textBox(country, width, height):

    COL = (0,0,0)
    
    covidinfo = covid.getCovidStats(country)
    travelString = WebScrape.getTravelRestrictions(country)
    destinationlist = destination.destinations(country)


    boxSurface = Surface((width,height), SRCALPHA)

    
    
    #title of country
    countryname = covidinfo['country']
    titlesurface = title_font.render(covidinfo['country'],False, COL)
    title_pos_center = (width//60,int(height//25))#title_pos_center = (width//3 - titlesurface.get_width()// 2,int(height//30))
    boxSurface.blit(titlesurface, title_pos_center)


    #track the current position of text
    curwidth = 4*width//30
    curheight = height//6+10
    
    #tourist entry

    boxSurface.blit(icon_tourist, (curwidth - icon_tourist.get_width() - 5, curheight))
    offset = blit_text(boxSurface, travelString, (curwidth, curheight), body_font, COL)
    curheight = offset + 15

    #covid rating

    boxSurface.blit(icon_covid, (curwidth - icon_tourist.get_width() - 5, curheight))
    
    rating = covidinfo['rating']
    cases = covidinfo['cases']

    covidstring = "{} is rated {}/10.0 for pandemic safety. (Active cases: {})".format(covidinfo['country'], rating, cases)
    offset = blit_text(boxSurface, covidstring, (curwidth, curheight), body_font, COL)
    curheight = offset + 15

    #destinations

    boxSurface.blit(icon_country, (curwidth - icon_tourist.get_width() - 5, curheight))
    offset = blit_text(boxSurface,"Destinations in {}".format(covidinfo['country']), (curwidth, curheight), body_font, COL)
    curheight = offset
    for i in range(len(destinationlist)):
        offset = blit_text(boxSurface, "{}. {}".format(i+1, destinationlist[i]), (curwidth, curheight), body_font, COL)
        curheight = offset

    

    



    return boxSurface


    
    
#tester code

if __name__ == "__main__":

    '''
    countryDict = {"Canada" : (255, 0, 0),
               "India" : (255, 138, 0),
               "USA": (52, 100, 235),
               "Mexico": (13, 107, 21),
               "Brazil": (10,209,27),
               "Spain": (235, 231, 30),
               "South Africa": (181,5,38),
               "Ethiopia": (27, 128, 135),
               "Egypt": (225, 227, 209),
               "Saudi Arabia": (3,37,84),
               "Russia": (134, 17,17),
               "Thailand": (88,3,110),
               "China": (189, 30, 57),
               "Australia": (9, 66, 181),
               "Algeria": (60,255,0),
               "Nigeria": (204, 0, 255)}
    '''

    countryDict = {"Canada" : (255, 0, 0),
               "India" : (255, 138, 0),
               "USA": (52, 100, 235),
               "Mexico": (13, 107, 21),
               "Brazil": (10,209,27),
               "Spain": (235, 231, 30),
               "South Africa": (181,5,38),
               "Ethiopia": (27, 128, 135),
               "Egypt": (225, 227, 209),
               "Saudi Arabia": (3,37,84),
               "Russia": (134, 17,17),
               "Thailand": (88,3,110),
               "China": (189, 30, 57),
               "Australia": (9, 66, 181)}

    alltextboxes = []
    
    for key in countryDict.keys():
        alltextboxes.append(textBox(key, 800,600))

    size=width,height = 1280,720
    screen=display.set_mode(size)

    
    

    
    clickable = True
    running=True

    count = 0
    delay = 200
    
    while running:
        for evt in event.get():
            if evt.type==QUIT:
                running=False

        #print(count)
        mpos=mouse.get_pos()
        mb=mouse.get_pressed()

        
        
        screen.fill((200,200,200))
        screen.blit(alltextboxes[count//delay], (100,100))
        
        count += 1
        count = 0 if count+2 > delay * len(countryDict.keys()) else count
        
        display.flip() 

    quit()