from random import randint
from sledo.generate.field_generators.base import FieldGenerator

values = ("Austria",
          "Belgium",
          "Bulgaria",
          "Croatia",
          "Cyprus",
          "Czech Republic",
          "Denmark",
          "Estonia",
          "Finland",
          "France",
          "Germany",
          "Greece",
          "Hungary",
          "Ireland",
          "Italy",
          "Latvia",
          "Lithuania",
          "Luxembourg",
          "Malta",
          "Netherlands",
          "Poland",
          "Portugal",
          "Romania",
          "Slovakia",
          "Slovenia",
          "Spain",
          "Sweden",
          "United States",
          "Japan",
          "United Kingdom",
          "Bangladesh",
          "Argentina",
          "China")

count = len(values) - 1


class CustomerAddressGenerator(FieldGenerator):
    def generate(self, **_):
        return values[randint(0, count)]

[((43, 22, 43, 39), 'random.randint', 'randint', ({(43, 30, 43, 31): '(0)', (43, 33, 43, 38): 'count'}, {}), '(0, count)', False, 'from random import randint\n')]