#!/usr/bin/env python3

"""

Draw country flags with the Python Turtle.

Have fun!

Peace, coolcornucopia.

https://github.com/coolcornucopia/

"""

#

# Python ressources:

# https://docs.python.org/3/library/turtle.html

#

# Wikipedia ressources:

# https://en.wikipedia.org/wiki/List_of_aspect_ratios_of_national_flags

# https://en.wikipedia.org/wiki/List_of_ISO_3166_country_codes

from turtle import Turtle, Screen, mainloop

import math

import locale

import unicodedata # Use to sort strings with accents, see strip_accents()

import time

### CONFIGURATION ###

KEY_EXIT = "Escape"

MOUSE_BUTTON_EXIT = 2 # mouse buttons: 1 left, 2 center, 3 right

MOUSE_BUTTON_NORMAL = 1

COUNTRY_NAMES_FILENAME = "country_names"

DEFAULT_LANGUAGE = "en"

fast_draw = True

FLAG_BORDER_COL = 'black'

FLAG_DEFAULT_RATIO = 2/3 # preferred flag ratio size between width & height

FULLSCREEN = True

DEBUG = False

#DEBUG = True

### GLOBAL VARIABLES ###

# This is our "current turtle (ct)"

ct = Turtle()

# TODO (resolution, default white background)

screen = Screen()

if FULLSCREEN:

screen.setup(width = 1.0, height = 1.0)

my_screenclicked = False

my_keypressed = False

country_names = {} # Content depends on the language

### DRAWING PRIMITIVES ###

# Useful fonction to move the pen and reset the turtle orientation

def prepare_drawing(x, y, rotation=0):

ct.pendown()

def rectangle(x, y, width, height):

ct.forward(height)

def rectangle_filled(x, y, width, height):

ct.begin_fill()

rectangle(x, y, width, height)

ct.end_fill()

def square(x, y, width):

rectangle(x, y, width, width)

def square_filled(x, y, width):

rectangle_filled(x, y, width, width)

# For the circle, use the diameter instead of the radius because it is

# then easier to make objects touch themselves, avoiding x2 in user code.

def circle(center_x, center_y, diameter):

# Move the circle center following Turtle circle() usage

prepare_drawing(center_x, center_y - diameter / 2)

ct.circle(diameter / 2)

def circle_filled(center_x, center_y, diameter):

ct.begin_fill()

circle(center_x, center_y, diameter)

ct.end_fill()

# The cross is inside a "width" diameter circle

def cross(center_x, center_y, width):

ct.forward(width)

# This five pointed star function draws a star "standing with arms open

# horizontally" and a span of "width" and returns the coordinates and sizes

# of the surrounding rectangle, it is then easier to align the star with

# other shapes. The drawing algorithm has been adapted from

# https://stackoverflow.com/questions/26356543/turtle-graphics-draw-a-star

# TODO better document rotation (clockwise here)

def five_pointed_star(center_x, center_y, width, rotation=0):

return center_x - d / 2, center_y + rc, d, h

# (read five_pointed_star() above function description for details)

def five_pointed_star_filled(center_x, center_y, width, rotation=0):

return x, y, w, h

def polygon(poly):

ct.goto(tuple(poly[0])) # close the polygon

def polygon_filled(poly):

ct.begin_fill()

polygon(poly)

ct.end_fill()

### HELPER FUNCTIONS FOR FLAGS DRAWING ###

# TODO better document below functions

def vertical_strips(x, y, width, height, *colors):

rectangle_filled(x + i * w, y, w, height)

def horizontal_strips(x, y, width, height, *colors):

rectangle_filled(x, y - i * h, width, h)

# TODO Document parameters

def rectangle_circle(x, y, width, height,

circle_filled(circ_center_x, circ_center_y, diameter)

# TODO Document parameters

def cross_filled(x, y, width, height,

rectangle_filled(x, y1, width, h)

def rectangle_filled_color(x, y, width, height, color):

ct.color(color)

rectangle_filled(x, y, width, height)

def circle_filled_color(center_x, center_y, diameter, color):

ct.color(color)

circle_filled(center_x, center_y, diameter)

def five_pointed_star_filled_color(center_x, center_y, width, color, rotation=0):

ct.color(color)

five_pointed_star_filled(center_x, center_y, width, rotation)

def polygon_filled_color(poly, color):

ct.color(color)

polygon_filled(poly)

def circle_coord(center_x, center_y, radius, angle_pc):

return x, y

### COUNTRY FLAG DRAWING FUNCTIONS ###

# Note Following functions do not take into account directly the

# aspect ratio and the border, so you can use them directly

# depending of your need... or via the class with proper aspect ratio :-)

def flag_Armenia(x, y, width, height):

horizontal_strips(x, y, width, height, '#D90012', '#0033A0', '#F2A800')

def flag_Austria(x, y, width, height):

horizontal_strips(x, y, width, height, '#ED2939', 'white', '#ED2939')

def flag_Bahamas(x, y, width, height):

horizontal_strips(x, y, width, height, '#00778B', '#FFC72C', '#00778B')

polygon_filled_color(((x, y), (x + width/2.3, y - height/2),

(x, y - height)), 'black')

def flag_Bangladesh(x, y, width, height):

rectangle_circle(x, y, width, height, 45/100, 1/2, 2/5,

'#006a4e', '#f42a41')

def flag_Belgium(x, y, width, height):

vertical_strips(x, y, width, height, 'black', '#FAE042', '#ED2939')

def flag_Benin(x, y, width, height):

horizontal_strips(x, y, width, height, '#FCD116', '#E8112D')

rectangle_filled_color(x, y, width / 2.5, height, '#008751')

def flag_Bolivia(x, y, width, height):

horizontal_strips(x, y, width, height, '#D52B1E', '#F9E300', '#007934')

# TODO Please finalize me

def flag_Botswana(x, y, width, height):

rectangle_filled_color(x, y, width, height, '#6DA9D2')

rectangle_filled_color(x, y - height * 3/8, width, height/4, 'white')

rectangle_filled_color(x, y - height/2.4, width, height/6, 'black')

def flag_Bulgaria(x, y, width, height):

horizontal_strips(x, y, width, height, 'white', '#00966E', '#D62612')

def flag_Cameroon(x, y, width, height):

vertical_strips(x, y, width, height, '#007A5E', '#CE1126', '#FCD116')

five_pointed_star_filled_color(x + width/2, y - height/2, width/6, '#FCD116')

def flag_China(x, y, width, height):

five_pointed_star_filled(x + width * 1/3, y - height * 9/20, ssw, 360-21)

def flag_Estonia(x, y, width, height):

horizontal_strips(x, y, width, height, '#0072ce', 'black', 'white')

def flag_Finland(x, y, width, height):

rectangle_filled_color(x, y, width, height, 'white')

cross_filled(x, y, width, height, 65/180, 1/2, 1/6, 3/11, '#003580')

def flag_France(x, y, width, height):

vertical_strips(x, y, width, height, '#002395', 'white', '#ED2939')

def flag_Gabon(x, y, width, height):

horizontal_strips(x, y, width, height, '#009e60', '#fcd116', '#3a75c4')

def flag_Gambia(x, y, width, height):

horizontal_strips(x, y, width, height, '#CE1126', 'white', '#3A7728')

rectangle_filled_color(x, y - height/2.57, width, height/4.5, '#0C1C8C')

def flag_Germany(x, y, width, height):

horizontal_strips(x, y, width, height, '#000', '#D00', '#FFCE00')

def flag_Greece(x, y, width, height):

1/13.5/0.37, 1/9/(5/9), w)

def flag_Iceland(x, y, width, height):

rectangle_filled_color(x, y, width, height, '#02529C')

cross_filled(x, y, width, height, 0.36, 1/2, 16/100, 2/9, 'white')

cross_filled(x, y, width, height, 0.36, 1/2, 8/100, 1/9, '#DC1E35')

def flag_India(x, y, width, height):

circle_filled_color(xx, yy, width * 10.5/1350, '#008')

def flag_Indonesia(x, y, width, height):

horizontal_strips(x, y, width, height, 'red', 'white')

def flag_Ivory_Coast(x, y, width, height):

vertical_strips(x, y, width, height, '#f77f00', 'white', '#009e60')

def flag_Japan(x, y, width, height):

rectangle_circle(x, y, width, height, 1/2, 1/2, 2/5,

'white', '#bc002d')

def flag_Myanmar(x, y, width, height):

five_pointed_star_filled_color(x + width / 2, y - height / 1.9, d, 'white')

def flag_Pakistan(x, y, width, height):

five_pointed_star_filled(x + width * 652/900, y - height * 216/600, width * 114/900, 23)

def flag_Russia(x, y, width, height):

horizontal_strips(x, y, width, height, 'white', '#0039A6', '#D52B1E')

def flag_Seychelles(x, y, width, height):

polygon_filled_color((bl, (xw, y - h * 2), (xw, y - height)), '#007a3d')

def flag_Somalia(x, y, width, height):

rectangle_filled_color(x, y, width, height, '#4189DD')

five_pointed_star_filled_color(x + width/2, y - height/2, width/3.28, 'white')

def flag_Sweden(x, y, width, height):

rectangle_filled_color(x, y, width, height, '#006AA7')

cross_filled(x, y, width, height, 3/8, 1/2, 1/8, 1/5, '#FECC00')

def flag_United_States(x, y, width, height):

star_y -= star_height

### FLAGS MANAGEMENT FUNCTIONS ###

class Flag(object):

self.drawing_func(x, y, width, width * self.ratio)

# Dictionnary of all the flags (the key is the flag drawing function)

flags_dict = dict()

flags_dict[flag_Armenia] = Flag( 51, 1/2 , flag_Armenia)

flags_dict[flag_Austria] = Flag( 40, 2/3 , flag_Austria)

flags_dict[flag_Bahamas] = Flag( 44, 1/2 , flag_Bahamas)

flags_dict[flag_Bangladesh] = Flag( 50, 3/5 , flag_Bangladesh)

flags_dict[flag_Belgium] = Flag( 56, 13/15, flag_Belgium)

flags_dict[flag_Benin] = Flag(204, 2/3 , flag_Benin)

flags_dict[flag_Bolivia] = Flag( 68, 15/22, flag_Bolivia)

flags_dict[flag_Botswana] = Flag( 72, 2/3 , flag_Botswana)

flags_dict[flag_Bulgaria] = Flag(100, 3/5 , flag_Bulgaria)

flags_dict[flag_Cameroon] = Flag(120, 2/3 , flag_Cameroon)

flags_dict[flag_China] = Flag(156, 2/3 , flag_China)

flags_dict[flag_Estonia] = Flag(233, 7/11, flag_Estonia)

flags_dict[flag_Finland] = Flag(246, 11/18, flag_Finland)

flags_dict[flag_France] = Flag(250, 2/3 , flag_France)

flags_dict[flag_Gabon] = Flag(266, 3/4 , flag_Gabon)

flags_dict[flag_Gambia] = Flag(270, 2/3 , flag_Gambia)

flags_dict[flag_Germany] = Flag(276, 3/5 , flag_Germany)

flags_dict[flag_Greece] = Flag(300, 2/3 , flag_Greece)

flags_dict[flag_Iceland] = Flag(352, 18/25, flag_Iceland)

flags_dict[flag_India] = Flag(356, 2/3 , flag_India)

flags_dict[flag_Indonesia] = Flag(360, 2/3 , flag_Indonesia)

flags_dict[flag_Ivory_Coast] = Flag(384, 2/3 , flag_Ivory_Coast)

flags_dict[flag_Japan] = Flag(392, 2/3 , flag_Japan)

flags_dict[flag_Myanmar] = Flag(104, 2/3 , flag_Myanmar)

flags_dict[flag_Pakistan] = Flag(586, 2/3 , flag_Pakistan)

flags_dict[flag_Russia] = Flag(643, 2/3 , flag_Russia)

flags_dict[flag_Seychelles] = Flag(690, 1/2 , flag_Seychelles)

flags_dict[flag_Somalia] = Flag(706, 2/3 , flag_Somalia)

flags_dict[flag_Sweden] = Flag(752, 5/8 , flag_Sweden)

flags_dict[flag_United_States] = Flag(840, 10/19, flag_United_States)

# Function to remove accents, useful for sorting, else "États-Unis" (fr)

# will be the last of the sorting list in French

# https://stackoverflow.com/questions/517923/what-is-the-best-way-to-remove-accents-in-a-python-unicode-string/518232#518232

def strip_accents(s):

return ''.join(c for c in unicodedata.normalize('NFD', s)

if unicodedata.category(c) != 'Mn')

# Sort the flags dictionnary by country code

sorted(flags_dict.items(), key=lambda x: x[1].country_code)

def draw_all_flags(width, border, country_names=False, ratio=False):

y -= width * 2/3 + border_inside # TODO 2/3 here is not so nice

# Helper function to avoid to test everytime if the country_names

# dictionnary is empty or a country code is missing

def get_country_name(code):

return name

def load_country_names(language):

return True

### EVENTS MANAGEMENT ###

def my_exit():

print("Bye")

screen.bye()

def my_exit_mouse(x, y):

#pylint: disable=unused-argument

my_exit()

def my_onscreenclick(x, y):

#print("my_screenclicked =", my_screenclicked, x, y)

def my_onkeypress():

global my_keypressed

my_keypressed = True

#print("my_keypressed =", my_keypressed)

def frame():

print("frame...")

#screen.update()

screen.ontimer(frame, 1000) # TODO hard coded value

def wait_click_or_key():

while not my_screenclicked and not my_keypressed:

screen.update()

time.sleep(0.5)

def install_event_management():

frame()

### SCREEN UPDATE HELPERS ###

# TODO rename me + test all parameters

def update_configure(fast=True, speed=2):

screen.tracer(True) # TODO not useful

def update_do():

global fast_draw

if fast_draw:

screen.update()

### TEST HELPERS ###

def test_primitives():

five_pointed_star_filled(40, -80, 40)

def test_flag(flag_function_name):

rectangle(-w/2, h/2, w, h)

def test_flag_class(flag_function_name, ratio=False):

rectangle(-w/2, h/2, w, h)

def screenshot(filename="screenshot"):

print("Screenshot done, filename =", filename)

### MAIN ###

def main():

return "Ready"

if __name__ == "__main__":

msg = main()

print(msg)

mainloop()