65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import sys
|
|
import os
|
|
|
|
import requests
|
|
import datetime
|
|
import arrow
|
|
from django.utils import timezone
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from rowers.models import Rower, Workout
|
|
import re
|
|
from django.db.models import Q
|
|
from rowers.dataprep import join_workouts
|
|
|
|
|
|
def name_short(name):
|
|
expr = '(.*)\s.*\(\d+\)'
|
|
match = re.findall(expr, name)
|
|
|
|
if match:
|
|
return match[0]
|
|
|
|
return name
|
|
|
|
|
|
def get_duplicates(a):
|
|
seen = {}
|
|
dupes = []
|
|
|
|
for x in a:
|
|
if x not in seen:
|
|
seen[x] = 1
|
|
else:
|
|
if seen[x] == 1:
|
|
dupes.append(x)
|
|
seen[x] += 1
|
|
|
|
return seen
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
rs = (r for r in Rower.objects.all() if r.ispaid and r.autojoin)
|
|
|
|
for r in rs:
|
|
workouts = Workout.objects.filter(user=r,
|
|
duplicate=False,
|
|
startdatetime__gte=timezone.now()-datetime.timedelta(days=2))
|
|
|
|
duplicates = get_duplicates(name_short(w.name) for w in workouts)
|
|
|
|
for name, count in duplicates.items():
|
|
if count > 1:
|
|
workouts2 = workouts.filter(
|
|
Q(name__contains=name))
|
|
|
|
duplicates2 = get_duplicates(w.date for w in workouts)
|
|
for dd, count in duplicates2.items():
|
|
if count > 1:
|
|
workouts3 = workouts2.filter(date=dd)
|
|
ids = [w.id for w in workouts3]
|
|
id, message = join_workouts(r, ids, title=name,
|
|
parent=workouts3[0],
|
|
killparents=True)
|