gamechanger updates (added events, teams). not used yet.

gamechanger/admin.py

@@ -1,8 +1,9 @@
 from django.contrib import admin
 
-from .models import Account, Player, Preferences
+from .models import Account, Player, Preferences, Team
 
 # Register your models here.
 admin.site.register(Account)
 admin.site.register(Preferences)
 admin.site.register(Player)
+admin.site.register(Team)

gamechanger/forms.py

@@ -5,14 +5,14 @@ from .models import Account, Player, Preferences
 
 
 class PreferencesForm(ModelForm):
+    season_id = ""
+
     class Meta:
         model = Preferences
-        fields = ["user", "season_id", "team_id"]
+        fields = ["user", "managed_team"]
         widgets = {
             "user": forms.HiddenInput(),
-            "managed_team_id": forms.TextInput(),
         }
-        labels = {"managed_team_id": "Selected Team"}
 
 
 class AccountForm(ModelForm):

gamechanger/migrations/0006_auto_20220615_1330.py

@@ -0,0 +1,37 @@
+# Generated by Django 3.2.13 on 2022-06-15 18:30
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gamechanger', '0005_alter_preferences_options'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Team',
+            fields=[
+                ('id', models.CharField(max_length=30, primary_key=True, serialize=False)),
+                ('season_id', models.CharField(max_length=30)),
+                ('name_slug', models.CharField(max_length=30)),
+                ('season_slug', models.CharField(max_length=30)),
+            ],
+        ),
+        migrations.RemoveField(
+            model_name='preferences',
+            name='season_id',
+        ),
+        migrations.RemoveField(
+            model_name='preferences',
+            name='team_id',
+        ),
+        migrations.AddField(
+            model_name='preferences',
+            name='managed_team',
+            field=models.OneToOneField(default=1, on_delete=django.db.models.deletion.CASCADE, to='gamechanger.team'),
+            preserve_default=False,
+        ),
+    ]

gamechanger/migrations/0007_remove_team_season_id.py

@@ -0,0 +1,17 @@
+# Generated by Django 3.2.13 on 2022-06-15 19:41
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gamechanger', '0006_auto_20220615_1330'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='team',
+            name='season_id',
+        ),
+    ]

gamechanger/migrations/0008_rename_name_slug_team_slug.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2022-06-15 21:18
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('gamechanger', '0007_remove_team_season_id'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='team',
+            old_name='name_slug',
+            new_name='slug',
+        ),
+    ]

gamechanger/models.py

@@ -14,12 +14,17 @@ class Account(models.Model):
     password = CharField(max_length=255)
 
 
+class Team(models.Model):
+    id = models.CharField(primary_key=True, max_length=30)
+    slug = CharField(max_length=30)
+    season_slug = CharField(max_length=30)
+
+
 class Preferences(models.Model):
     user = models.OneToOneField(
         User, on_delete=models.CASCADE, related_name="gamechanger_preferences"
     )
-    season_id = CharField(max_length=255)
+    managed_team = models.OneToOneField(Team, on_delete=models.CASCADE)
-    team_id = CharField(max_length=255)
 
     class Meta:
         verbose_name_plural = "preferences"

gamechanger/utils/gamechanger.py

@@ -7,7 +7,7 @@ import pytz
 import requests
 from bs4 import BeautifulSoup
 
-url = "https://gc.com/t/{season_id}/{team_id}/{page}"
+url = "https://gc.com/t/{season_id}/{team_slug}-{team_id}/{page}"
 
 
 def get_authenticated_session(request):
@@ -35,12 +35,16 @@ def get_authenticated_session(request):
 
 def submit_lineup(request, lineup):
     authenticated_session = get_authenticated_session(request)
-    season_id = request.user.gamechanger_preferences.season_id
+    season_id = request.user.gamechanger_preferences.managed_team.season_slug
-    team_id = request.user.gamechanger_preferences.team_id
+    team_slug = request.user.gamechanger_preferences.managed_team.slug
+    team_id = request.user.gamechanger_preferences.managed_team.id
     authenticated_session.headers.update(
         {
             "referer": url.format(
-                season_id=season_id, team_id=team_id, page="lineup_edit"
+                season_id=season_id,
+                team_slug=team_slug,
+                team_id=team_id,
+                page="lineup_edit",
             ),
             "x-csrftoken": authenticated_session.cookies.get("csrftoken"),
             "Content-Type": "application/x-www-form-urlencoded;",
@@ -48,12 +52,10 @@ def submit_lineup(request, lineup):
     )
     r = authenticated_session.post(
         cookies=authenticated_session.cookies,
-        url="https://gc.com/do-save-lineup/{team_id}".format(
+        url=f"https://gc.com/do-save-lineup/{team_id}",
-            team_id=team_id.split("-").pop()
-        ),
         json={"lineup": lineup},
     )
-    if r.status_code == 200:
+    if r.status_code == 20 and r.content == b"OK":
         return r
     else:
         raise requests.exceptions.RequestException(
@@ -61,8 +63,10 @@ def submit_lineup(request, lineup):
         )
 
 
-def scrape_page(season_id, team_id, page):
+def scrape_page(season_id, team_id, team_slug, page):
-    r = requests.get(url.format(season_id=season_id, team_id=team_id, page=page))
+    r = requests.get(
+        url.format(season_id=season_id, team_id=team_id, team_slug=team_slug, page=page)
+    )
     initialize_page_json = re.search(
         r'page.initialize\(\$.parseJSON\("(.*?)"\)', r.content.decode("unicode_escape")
     )
@@ -160,14 +164,34 @@ def stream():
 
 def stats(request):
     authenticated_session = get_authenticated_session(request)
-    season_id = request.user.gamechanger_preferences.season_id
+    season_id = request.user.gamechanger_preferences.managed_team.season_slug
-    team_id = request.user.gamechanger_preferences.team_id
+    team_id = request.user.gamechanger_preferences.managed_team.id
+    team_slug = request.user.gamechanger_preferences.managed_team.slug
     page = "stats/batting/Qualified/standard/csv"
+    authenticated_session.headers.update(
+        {
+            "referer": url.format(
+                season_id=season_id, team_id=team_id, team_slug=team_slug, page="stats"
+            ),
+            "x-csrftoken": authenticated_session.cookies.get("csrftoken"),
+        }
+    )
     r = authenticated_session.get(
-        url.format(season_id=season_id, team_id=team_id, page=page)
+        cookies=authenticated_session.cookies,
+        url=url.format(
+            season_id=season_id, team_id=team_id, team_slug=team_slug, page=page
+        ),
     )
 
-    roster = scrape_page(season_id, team_id, "roster")
+    if (
+        r.status_code != 200
+        or "Please sign in or join to continue." in r.content.decode("utf-8")
+    ):
+        raise Exception("Stats fetch failed.")
+
+    roster = scrape_page(
+        season_id=season_id, team_id=team_id, team_slug=team_slug, page="roster"
+    )
     id_lookup = {
         (p.get("fname"), p.get("lname")): p.get("player_id") for p in roster["roster"]
     }

gamechanger/views.py

@@ -1,3 +1,4 @@
+from django import forms
 from django.http import HttpResponse, HttpResponseNotAllowed, HttpResponseServerError
 from django.shortcuts import render
 from django.views.generic.edit import FormView
@@ -5,7 +6,7 @@ from django.views.generic.edit import FormView
 from teamsnap.views import get_teamsnap_client
 
 from .forms import AccountForm, PlayerFormSet, PreferencesForm
-from .models import Account, Player, Preferences
+from .models import Account, Player, Preferences, Team
 from .utils import gamechanger
 
 
@@ -30,10 +31,15 @@ class PreferencesFormView(FormView):
     def form_valid(self, form):
         # This method is called when valid form data has been POSTed.
         # It should return an HttpResponse.
-        if form.data["user"] == str(self.request.user.id):
+        if form.cleaned_data["user"].username == str(
+            self.request.user.gamechanger_account.user
+        ):
             form.save()
         return super().form_valid(form)
 
+    def form_invalid(self, form):
+        pass
+
     def get_initial(self):
         """
         Returns the initial data to use for forms on this view.
@@ -51,11 +57,32 @@ class PreferencesFormView(FormView):
         """
 
         try:
-            contact = Preferences.objects.get(user=self.request.user)
+            preferences = Preferences.objects.get(user=self.request.user)
-            form = PreferencesForm(instance=contact, **self.get_form_kwargs())
+            form = PreferencesForm(instance=preferences, **self.get_form_kwargs())
+
         except Preferences.DoesNotExist:
             form = super().get_form(self.form_class)
 
+        gc_session = gamechanger.get_authenticated_session(self.request)
+        teams = gamechanger.get_teams(gc_session)
+
+        team_instances = []
+        choices = []
+        for team in teams:
+            instance, _ = Team.objects.get_or_create(
+                id=team["id"],
+                slug="-".join(team["team_slug"].split("-")[:-1]),
+                season_slug=team["season_slug"],
+            )
+            team_instances.append(instance)
+            choices.append((team["id"], f"{team['name']} ({team['season']})"))
+
+        form.fields["managed_team"].widget = forms.Select(
+            choices=choices, attrs={"class": "form-control"}
+        )
+        # form.fields["managed_team"].choices = [choice[0] for choice in choices]
+        # form.fields["managed_team"].widget.choices = choices
+
         return form
 
 
@@ -97,8 +124,8 @@ class AccountFormView(FormView):
 
 
 def roster(request):
-    season_id = request.user.gamechanger_preferences.season_id
+    season_id = request.user.gamechanger_preferences.managed_team.season_slug
-    team_id = request.user.gamechanger_preferences.team_id
+    team_id = request.user.gamechanger_preferences.id
     page = "roster"
     d = gamechanger.scrape_page(season_id, team_id, page)
     roster = d["roster"]
@@ -110,8 +137,9 @@ def roster_import(request):
         from pyteamsnap.api import Member
 
         client = get_teamsnap_client(request)
-        season_id = request.user.gamechanger_preferences.season_id
+        season_id = request.user.gamechanger_preferences.managed_team.season_slug
-        team_id = request.user.gamechanger_preferences.team_id
+        team_slug = request.user.gamechanger_preferences.managed_team.slug
+        team_id = request.user.gamechanger_preferences.managed_team.id
         teamsnap_team_id = request.user.teamsnap_preferences.managed_team_id
         teamsnap_members = {
             f"{member.data['first_name']} {member.data['last_name']}": member
@@ -120,7 +148,7 @@ def roster_import(request):
 
         page = "roster"
 
-        d = gamechanger.scrape_page(season_id, team_id, page)
+        d = gamechanger.scrape_page(season_id, team_id, team_slug, page)
         roster = d["roster"]
         initial = [
             {
@@ -128,12 +156,18 @@ def roster_import(request):
                 "fname": player["fname"],
                 "lname": player["lname"],
                 "teamsnap_name": "{first_name} {last_name}".format(
-                    **teamsnap_members[f"{player['fname']} {player['lname']}"].data
+                    **getattr(
+                        teamsnap_members.get(f"{player['fname']} {player['lname']}"),
+                        "data",
+                        {"first_name": "", "last_name": ""},
+                    )
                 ),
                 "id": player.get("player_id"),
-                "teamsnap_member_id": teamsnap_members[
+                "teamsnap_member_id": getattr(
-                    f"{player['fname']} {player['lname']}"
+                    teamsnap_members.get(f"{player['fname']} {player['lname']}"),
-                ].data["id"],
+                    "data",
+                    {"id": None},
+                )["id"],
             }
             for player in roster
         ]