def create_test_data(self): user = User.get_or_create(123) user.admin = True User.get_or_create(12345) db.session.commit() team1 = Team.create(user, 'EnvyUs', 'EnvyUs', 'fr', 'nv', ['76561198053858673']) team2 = Team.create(user, 'Fnatic', 'Fnatic', 'se', 'fntc', ['76561198053858673']) server = GameServer.create(user, 'myserver1', '127.0.0.1', '27015', 'password', False) server.in_use = True GameServer.create(user, 'myserver2', '127.0.0.1', '27016', 'password', True) db.session.commit() season = Season.create( user, 'Season One Test', datetime.datetime.utcnow(), datetime.datetime.utcnow() + datetime.timedelta(days=1)) db.session.commit() Match.create(user, team1.id, team2.id, '', '', 1, False, 'Map {MAPNUMBER}', ['de_dust2', 'de_cache', 'de_mirage'], season.id, 'always_knife', 'CT', server.id) db.session.commit() vetoBan = Veto.create(1, 'EnvyUs', 'de_dust2', 'ban') vetoPick = Veto.create(1, 'EnvyUs', 'de_overpass', 'pick') db.session.commit()
def season_create(): if not g.user: return redirect('/login') form = SeasonForm(request.form) if request.method == 'POST': num_seasons = g.user.seasons.count() max_seasons = config_setting('USER_MAX_SEASONS') if max_seasons >= 0 and num_seasons >= max_seasons and not (util.is_admin(g.user) or util.is_super_admin(g.user)): flash('You already have the maximum number of seasons ({}) created'.format( num_seasons)) elif form.validate(): season = Season.create( g.user, form.data['season_title'], form.data['start_date'], form.data['end_date']) db.session.commit() app.logger.info('User {} created season {}' .format(g.user.id, season.id)) return redirect('/myseasons') else: get5.flash_errors(form) return render_template( 'season_create.html', form=form, user=g.user)