<commit_before>from django.core.management.base import BaseCommand
from django.db import transaction

from sampledatahelper.helper import SampleDataHelper

from cinemair.users.models import User
from cinemair.cinemas.models import Cinema
from cinemair.movies.models import Movie
from cinemair.shows.models import Show
from cinemair.events.models import Event

NUM_USERS = 9
NUM_SHOWS = 400
RANGE_EVENTS = [3, 8]
MOVIES_IDS = [157336, 155, 98, 76341, 278, 238, 600, 78, 935, 185, 539, 346, 289, 389, 914, 5156, 653, 626, 19]

CINEMAS = [
    ("The Urban Beach Cinema Conde Duque", "Conde Duque, 11"),
    ("Sunset Cinema", "Plaza de Cibeles"),
    ("Parque de la Bombilla", "Av. De Valladolid"),
    ("Auditorio Parque Calero", "<NAME>"),
    ("Cine Doré", "Santa Isabel, 3"),
    ("La Casa Encendida", "Ronda de Valencia, 2"),
    ("Cine en el paraíso", "Parque el Paraíso"),
]

class Command(BaseCommand):
    sd = SampleDataHelper(seed=1234567890)

    @transaction.atomic
    def handle(self, *args, **options):
        # Create admin user
        self._create_admin_user()

        # Create cinemas
        for name, address in CINEMAS:
            self._create_cinema(name, address)

        # Create movies
        for i in MOVIES_IDS:
            self._create_movie(i)

        # Create shows
        for i in range(1, NUM_SHOWS + 1):
            self._create_show()
<commit_msg>Add actual malasana event dates<commit_after>from django.core.management.base import BaseCommand
from django.db import transaction

from sampledatahelper.helper import SampleDataHelper

from cinemair.users.models import User
from cinemair.cinemas.models import Cinema
from cinemair.movies.models import Movie
from cinemair.shows.models import Show
from cinemair.events.models import Event

NUM_USERS = 9
NUM_SHOWS = 400
RANGE_EVENTS = [3, 8]
MOVIES_IDS = [157336, 155, 98, 76341, 278, 238, 600, 78, 935, 185, 539, 346, 289, 389, 914, 5156, 653, 626, 19, 11042, 773, 20607, 575, 31175]

CINEMAS = [
    ("Cine de verano de Malasaña", "Cal<NAME>, 8"),
    ("The Urban Beach Cinema Conde Duque", "Conde Duque, 11"),
    ("Sunset Cinema", "Plaza de Cibeles"),
    ("Parque de la Bombilla", "Av. De Valladolid"),
    ("Auditorio Parque Calero", "Calle José del Hierro"),
    ("Cine Doré", "Santa Isabel, 3"),
    ("La Casa Encendida", "Ronda de Valencia, 2"),
    ("Cine en el paraíso", "Parque el Paraíso"),
]

class Command(BaseCommand):
    sd = SampleDataHelper(seed=1234567890)

    @transaction.atomic
    def handle(self, *args, **options):
        # Create admin user
        self._create_admin_user()

        # Create cinemas
        for name, address in CINEMAS:
            self._create_cinema(name, address)

        # Create movies
        for i in MOVIES_IDS:
            self._create_movie(i)

        # Create shows
        for i in range(1, NUM_SHOWS + 1):
            self._create_show()

{'fraction_non_alphanumeric': 0.11011080332409973, 'fraction_numerical': 0.06440443213296398, 'mean_word_length': 3.2863501483679527, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3388655', 'n_tokens_mistral': 1131, 'n_tokens_neox': 942, 'n_words': 303}