def test_teamCIOsearch(self):
nation_table= nation_team_table.load()
self.assertEqual(teamCIOsearch(nation_table,"AAA"),0)
self.assertEqual(teamCIOsearch(nation_table,"AFG"),0)
self.assertEqual(teamCIOsearch(nation_table,"ALB"),1)
self.assertTrue(teamCIOsearch(nation_table,"MGL")!=0)
def test_find_national_team1(self):
man_or_woman = u'woman'
force_nation_team = False
time_of_race = pywikibot.WbTime(site=site,
year=2020,
month=8,
day=27,
precision='day')
year = time_of_race.year
log = Log()
nation_table = nation_team_table.load()
result_dic = {
'rank': [-1, 0, ''],
'last name': [-1, 1, ''],
'first name': [-1, 2, ''],
'name': [-1, 3, ''],
'result': [-1, 4, 'time'], #startlist only with time
'points': [-1, 5, 'points'],
'team code': [-1, 7, ''],
'ecart': [1, 6, 'time'], #always created
'bib': [-1, 8, ''] #dossard
}
result_table, row_count, ecart = table_reader('National_team_tests',
result_dic, 0, False)
result_table = sorted(result_table, key=lambda tup: int(tup[8]))
list_of_cyclists, all_riders_found, log, list_of_teams, all_teams_found = cyclists_table_reader(
pywikibot, site, repo, result_table, result_dic, nosortkey=True)
res_list_of_cyclists, log = find_national_team(
pywikibot, site, repo, list_of_cyclists, result_table, result_dic,
row_count, nation_table, force_nation_team, year, log,
man_or_woman, time_of_race)
#NED makes problem
self.assertEqual(res_list_of_cyclists[9].national_team,
True) #Fra then ITA
self.assertEqual(res_list_of_cyclists[10].national_team, True)
self.assertEqual(res_list_of_cyclists[11].national_team, True)
self.assertEqual(res_list_of_cyclists[12].national_team, True)
self.assertEqual(res_list_of_cyclists[13].national_team, True)
self.assertEqual(res_list_of_cyclists[14].national_team, True)
self.assertEqual(res_list_of_cyclists[15].national_team, True)
self.assertEqual(res_list_of_cyclists[16].national_team, True)
self.assertEqual(res_list_of_cyclists[17].national_team, True)
self.assertEqual(res_list_of_cyclists[18].national_team, True)
self.assertEqual(res_list_of_cyclists[19].national_team, True)
self.assertEqual(res_list_of_cyclists[20].national_team, True)
def test_find_national_team2(self):
man_or_woman = u'woman'
time_of_race = pywikibot.WbTime(site=site,
year=2020,
month=8,
day=27,
precision='day')
year = time_of_race.year
log = Log()
nation_table = nation_team_table.load()
result_dic = {
'rank': [-1, 0, ''],
'last name': [-1, 1, ''],
'first name': [-1, 2, ''],
'name': [-1, 3, ''],
'result': [-1, 4, 'time'], #startlist only with time
'points': [-1, 5, 'points'],
'team code': [-1, 7, ''],
'ecart': [1, 6, 'time'], #always created
'bib': [-1, 8, ''] #dossard
}
force_nation_team = True #Test force_nation_team
result_table, row_count, ecart = table_reader(
'National_team_tests_orig', result_dic, 0, False)
result_table = sorted(result_table, key=lambda tup: int(tup[8]))
list_of_cyclists, all_riders_found, log, list_of_teams, all_teams_found = cyclists_table_reader(
pywikibot, site, repo, result_table, result_dic, nosortkey=True)
res_list_of_cyclists, log = find_national_team(
pywikibot, site, repo, list_of_cyclists, result_table, result_dic,
row_count, nation_table, force_nation_team, year, log,
man_or_woman, time_of_race)
self.assertEqual(res_list_of_cyclists[0].team, 'Q74847203')
self.assertEqual(res_list_of_cyclists[7].team, 'Q74847203')
self.assertEqual(res_list_of_cyclists[8].team, 'Q74749584')
self.assertEqual(res_list_of_cyclists[9].team, 'Q74749584')
def cycling_init_bot():
nation_table = nation_team_table.load()
selector = 4
#0-4: init the year
#5-6: sorter
#7-8: create races
#9-10: complete the ranking and start list
#11-15: others
if selector == 0:
from src import national_team_creator
man_or_woman = u'man'
start_year = 2022
end_year = 2022
#optional
country = False #u'FRA' #false is not
national_team_creator.f(pywikibot,
site,
repo,
nation_table,
man_or_woman,
start_year,
end_year,
country=country)
elif selector == 1:
from src import national_championship_creator
from src.data import cc_table
#cc
cc_table = cc_table.load()
man_or_woman = u'both' #both
option = u'clmon' #'clmoff'
start_year = 2022
end_year = 2022
CC = True
national_championship_creator.f(pywikibot, site, repo, cc_table,
man_or_woman, option, start_year,
end_year, CC)
elif selector == 2: #
#not cc
from src import national_championship_creator
man_or_woman = u'both'
option = u'clmon' #'clmoff'
start_year = 2022
end_year = 2022
country = False #optional/False u'ESA'
CC = False
national_championship_creator.f(pywikibot,
site,
repo,
nation_table,
man_or_woman,
option,
start_year,
end_year,
CC,
country=country)
elif selector == 3:
from src import calendar_importer
from src.data import race_list
race_table, race_dic = race_list.load()
test = False
man_or_woman = u'woman'
filename = 'calendrier2022'
year = 2022
calendar_importer.f(pywikibot, site, repo, nation_table, test,
race_table, race_dic, man_or_woman, filename, year)
elif selector == 4:
#more details in the table with activate and group
from src import team_creator
from src.data import pro_team_table, amateur_team_table
pro_or_amateur = 1 #1 is pro
year = 2022
prov = True
category_id = 2466826
"""
UCI WorldTeam (Q6154783)
UCI ProTeam (Q78464255)
UCI Continental Team (Q1756006)
UCI Women’s WorldTeam (Q80425135)
WOmen conti Q2466826
Female amateur cycling team (Q26849121)
Amateur cycling team (Q20652655)
"""
if prov:
team_table = [[0 for x in range(7)] for y in range(2)]
team_table[1][1] = u"Colombia Tierra De Atletas-GW-Shimano"
team_table[1][2] = 110794483 #master
team_table[1][3] = u'COL' #country
team_table[1][4] = u'CTF' #UCI code
team_table[1][5] = 2 #don't modify
team_table[1][6] = 1 #don't modify
_, team_dic = pro_team_table.load()
else:
if pro_or_amateur == 1:
team_table, team_dic = pro_team_table.load()
else:
team_table, team_dic = amateur_team_table.load()
team_creator.f(pywikibot,
site,
repo,
team_table,
nation_table,
team_dic,
year,
category_id=category_id)
elif selector == 5:
from src import sorter
id_team = u'Q105648394'
# 'has part (P527)', 'participating team (P1923)'
prop = "P527"
test = False
sorter.name_sorter(pywikibot, site, repo, id_team, prop, test)
elif selector == 6:
from src import sorter
id_team = u'Q107093611'
# if victory: property_number = 2522 # victoire
# else: # property_number = 527 # comprend
prop = "P527"
test = False
sorter.date_sorter(pywikibot, site, repo, id_team, prop, test)
elif selector == 7:
from src import race_creator
race_name = u"Watersley Womens Challenge"
id_race_master = 79032687
create_stages = True
year = 2021
race_begin = pywikibot.WbTime(site=site,
year=year,
month=9,
day=3,
precision='day')
end_date = pywikibot.WbTime(site=site,
year=year,
month=9,
day=5,
precision='day')
first_stage = 1
last_stage = 3
countryCIO = u'NED'
classe = '2.2'
edition_nr = '1'
single_race = False
man_or_woman = u'woman'
race_creator.f(pywikibot,
site,
repo,
nation_table,
race_name,
single_race,
man_or_woman,
id_race_master=id_race_master,
countryCIO=countryCIO,
classe=classe,
race_begin=race_begin,
edition_nr=edition_nr,
end_date=end_date,
only_stages=False,
create_stages=create_stages,
first_stage=first_stage,
last_stage=last_stage,
year=year)
elif selector == 8:
from src import race_creator
race_name = u"Binche-Chimay-Binche féminin"
id_race_master = 108914692
year = 2021
race_date = pywikibot.WbTime(site=site,
year=year,
month=10,
day=5,
precision='day')
countryCIO = u'BEL'
classe = '1.2'
edition_nr = '1'
single_race = True
man_or_woman = u'woman'
race_creator.f(pywikibot,
site,
repo,
nation_table,
race_name,
single_race,
man_or_woman,
race_begin=race_date,
edition_nr=edition_nr,
id_race_master=id_race_master,
countryCIO=countryCIO,
classe=classe,
year=year)
elif selector == 9:
from src import classification_importer
id_race = 'Q103982710'
stage_or_general = 6 # 1 == stage, #0 == general, #2 == point, #3 mountains,#4 youth, #5 team, #6 team ponts, #7 youth points
#8 == sprints
final = False
maxkk = 10
year = 2021 #for team
startliston = True
test = False
man_or_woman = u'woman'
classification_importer.f(pywikibot,
site,
repo,
stage_or_general,
id_race,
final,
maxkk,
test,
year=year,
startliston=startliston,
man_or_woman=man_or_woman)
elif selector == 10:
from src import startlist_importer
id_race = 'Q104129401'
prologue_or_final = 2 #0=prologue, 1=final, 2=one day race
chrono = False
test = False
man_or_woman = u'woman'
force_nation_team = False
time_of_race = pywikibot.WbTime(site=site,
year=2021,
month=10,
day=23,
precision='day')
startlist_importer.f(pywikibot, site, repo, prologue_or_final, id_race,
time_of_race, chrono, test, nation_table,
man_or_woman, force_nation_team)
elif selector == 12:
from src import rider_fast_init
name = u"Courtney Sherwell"
countryCIO = u'AUS'
man_or_woman = u'woman'
rider_fast_init.f(pywikibot, site, repo, nation_table, name,
countryCIO, man_or_woman)
elif selector == 13:
from src import champ_list_creator
man_or_woman = u'woman'
actualize = True
start_year = 2021
champ_list_creator.f(pywikibot, site, repo, man_or_woman, start_year,
actualize)
elif selector == 14:
from src import uci_classification
man_or_woman = u'man'
id_master_UCI = u'Q104218422' #Q97367360
year = u'2021'
filename = u'OceaniaRanking2021man' #'UCIranking' CIranking2020man
test = False
cleaner = False #delete the UCI ranking
UCIranking = False #for team team
bypass = True #don't interrupt if not all riders found
uci_classification.f(pywikibot, site, repo, year, id_master_UCI,
filename, cleaner, test, man_or_woman, UCIranking,
bypass)
elif selector == 15:
from src import cycling_init_bot_low as low
id_item = u'Q57267790'
property_nummer = "P3496"
low.delete_property(pywikibot, repo, id_item, property_nummer)
elif selector == 16:
#only stage
from src import race_creator
#race_name=u"Grand Prix Mediterrennean"
# id_race_master='104640104'
#year=2021
stage_race_id = "Q104640104" #only for onlystages
first_stage = 0
last_stage = 3
#single_race=True
man_or_woman = u'woman'
race_creator.f(
pywikibot,
site,
repo,
nation_table,
None, #race_name,
False, # single_race
man_or_woman,
stage_race_id=stage_race_id,
only_stages=True,
first_stage=first_stage,
last_stage=last_stage)
#year=year)
elif selector == 17:
from src import get_rider_tricot
id_race = 'Q65956705'
chrono = False
test = False
man_or_woman = u'woman'
time_of_race = pywikibot.WbTime(site=site,
year=2021,
month=7,
day=31,
precision='day')
get_rider_tricot.scan(pywikibot, site, repo, id_race, time_of_race,
chrono, test, man_or_woman)
elif selector == 18:
from src import get_rider_tricot
id_race = 'Q101426092'
chrono = True
test = False
man_or_woman = u'woman'
get_rider_tricot.scan_existing(pywikibot, site, repo, id_race, chrono,
test, man_or_woman)
else:
print('do nothing')
def test_IDtoCIOsearch(self):
nation_table= nation_team_table.load()
self.assertEqual(IDtoCIOsearch(nation_table,456789),"no code")
self.assertEqual(IDtoCIOsearch(nation_table,889),"AFG")
self.assertEqual(IDtoCIOsearch(nation_table,804),"PAN")
def test_CIOtoIDsearch(self):
team_table=nation_team_table.load()
self.assertEqual(CIOtoIDsearch(team_table, "FRA") , 142)
self.assertEqual(CIOtoIDsearch(team_table, "GBR") , 145)
self.assertEqual(CIOtoIDsearch(team_table, "ZZZ") , 0)