async def test_taskdownload_refresh_api(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
soarscore.tasks = []
await act.wait_for_tasks()
rendered = wtb.render()
assert "No task for today" in rendered
# When refresh() is called
soarscore.tasks = [
SoarScoreTaskInfo(
comp_class="Club",
title="Club Task",
day_no=1,
task_no=1,
timestamp="now",
task_url="http://soarscore.com/club.tsk",
),
]
wdg.refresh()
# Then view should be updated - no todays task in this case
await act.wait_for_tasks()
rendered = wtb.render()
assert "Club Task day 1 task 1" in rendered
async def test_taskdownload_no_ss_url(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition("test", "Test Competition",)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
# When soaringspot url is missing, do not fetch current tasks
rendered = wtb.render()
assert rendered.strip() == ""
async def test_taskdownload_task_download(
storage_dir, soarscore, widget_testbed
) -> None:
state = {"downloading_task": None}
def _on_download_task(ev, task):
state["downloading_task"] = task
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
urwid.connect_signal(wdg, "download", _on_download_task)
wtb = widget_testbed.for_widget(wdg)
soarscore.tasks = [
SoarScoreTaskInfo(
comp_class="Club",
title="Club Task",
day_no=1,
task_no=1,
timestamp="now",
task_url="http://soarscore.com/club.tsk",
),
SoarScoreTaskInfo(
comp_class="Standard",
title="Standard Task",
day_no=1,
task_no=1,
timestamp="now",
task_url="http://soarscore.com/standard.tsk",
),
]
await act.wait_for_tasks()
rendered = wtb.render()
assert "Club Task day 1 task 1" in rendered
# When we click download
focused = wtb.get_focus_widgets()[-1]
assert focused.label == "Download"
await wtb.keypress("enter")
# "download" event emits
assert state["downloading_task"] is not None
tsk = state["downloading_task"]
assert tsk.title == "Club Task"
async def test_taskdownload_notask(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
await act.wait_for_tasks()
rendered = wtb.render()
assert "No task for today" in rendered
assert "Refresh" in rendered
async def test_taskdownload_noclasses(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test", "Test Competition", soaringspot_url="http://soaringspot.com/test"
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
# await act.wait_for_tasks()
await asyncio.sleep(0)
assert "Fetching today's task..." in wtb.render()
# When classes are not known, we cannot pick task
await act.wait_for_tasks()
assert "Select competition class to download task" in wtb.render()
async def test_taskdownload_refresh_btn(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
soarscore.tasks = [
SoarScoreTaskInfo(
comp_class="Club",
title="Old Club Task",
day_no=1,
task_no=1,
timestamp="yesteday",
task_url="http://soarscore.com/club.tsk",
),
]
await act.wait_for_tasks()
rendered = wtb.render()
assert "Old Club Task day 1 task 1" in rendered
assert "Refresh" in rendered
# When refresh button is pressed
soarscore.tasks = [
SoarScoreTaskInfo(
comp_class="Club",
title="New Club Task",
day_no=2,
task_no=2,
timestamp="now",
task_url="http://soarscore.com/club.tsk",
),
]
await wtb.keypress("right", "enter")
await act.wait_for_tasks()
rendered = wtb.render()
assert "New Club Task day 2 task 2" in rendered
async def test_taskdownload_task_view(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
soarscore.tasks = [
SoarScoreTaskInfo(
comp_class="Club",
title="Club Task",
day_no=1,
task_no=1,
timestamp="now",
task_url="http://soarscore.com/club.tsk",
),
SoarScoreTaskInfo(
comp_class="Standard",
title="Standard Task",
day_no=1,
task_no=1,
timestamp="now",
task_url="http://soarscore.com/standard.tsk",
),
]
# Task title and other data should be displayed
await act.wait_for_tasks()
rendered = wtb.render()
assert "Today's task" in rendered
assert "Club Task day 1 task 1" in rendered
assert "Generated on now" in rendered
assert "Download" in rendered
assert "Refresh" in rendered
async def test_taskdownload_fetch_error(storage_dir, soarscore, widget_testbed) -> None:
comp = storage.StoredCompetition(
"test",
"Test Competition",
soaringspot_url="http://soaringspot.com/test",
selected_class="Club",
)
act = ActivityStub(urwid.SolidFill("T"))
wdg = TaskDownloadWidget(act, comp)
wtb = widget_testbed.for_widget(wdg)
soarscore.tasks = []
await act.wait_for_tasks()
rendered = wtb.render()
assert "No task for today" in rendered
assert "Refresh" in rendered
soarscore.fetch_latest_tasks_exc = SoarScoreClientError("Internet error")
focused = wtb.get_focus_widgets()[-1]
assert focused.label == " Refresh "
await wtb.keypress("enter")
await act.wait_for_tasks()
assert "Error fetching today's task: Internet error" in wtb.render()
def create_view(self) -> urwid.Widget:
p2 = lambda w: urwid.Padding(w, left=2)
self.class_widget = CompetitionClassSelectorWidget(self, self.competition)
urwid.connect_signal(self.class_widget, "change", self._on_class_changed)
self.task_widget = TaskDownloadWidget(self, self.competition)
urwid.connect_signal(self.task_widget, "download", self._on_download_task)
self.airspace_group: List[urwid.Widget] = []
self.airspace_pile = urwid.Pile(
[
self._make_file_radio(
sf,
self.airspace_group,
sf.name == self.competition.airspace,
self._on_airspace_changed,
)
for sf in self.airspaces
]
)
self.waypoint_group: List[urwid.Widget] = []
self.waypoint_pile = urwid.Pile(
[
self._make_file_radio(
sf,
self.waypoint_group,
sf.name == self.competition.waypoints,
self._on_waypoint_changed,
)
for sf in self.waypoints
]
)
self.profile_pile = urwid.Pile(
[self._make_profile_checkbox(prf) for prf in self.profiles]
)
self.download_status = urwid.Text("")
form = urwid.Pile(
[
self.class_widget,
urwid.Divider(),
self.task_widget,
urwid.Divider(),
self.download_status,
urwid.Divider(),
urwid.Text("Airspace files"),
p2(self.airspace_pile),
urwid.Divider(),
urwid.Text("Waypoint files"),
p2(self.waypoint_pile),
urwid.Divider(),
urwid.Text("XCSoar profiles"),
p2(self.profile_pile),
urwid.Divider(),
self._create_buttons(),
urwid.Divider(),
self._create_credits(),
]
)
filler = urwid.Filler(form, valign=urwid.TOP)
self.status = widget.CMFlashMessage(self)
return urwid.Frame(
widget.CMScreenPadding(filler),
header=widget.CMScreenHeader(self.competition.title),
footer=widget.CMScreenPadding(self.status),
)
class CompetitionDetailsScreen(Activity):
competition: storage.StoredCompetition
def show(self) -> None:
cid = storage.get_settings().current_competition_id
assert cid is not None
comp = storage.load_competition(cid)
assert comp is not None
self.competition = comp
self.airspaces = storage.get_airspace_files(cid)
self.waypoints = storage.get_waypoint_files(cid)
self.profiles = xcsoar.list_xcsoar_profiles()
super().show()
self._flashtask = None
self.async_task(self._update_competition_files())
def create_view(self) -> urwid.Widget:
p2 = lambda w: urwid.Padding(w, left=2)
self.class_widget = CompetitionClassSelectorWidget(self, self.competition)
urwid.connect_signal(self.class_widget, "change", self._on_class_changed)
self.task_widget = TaskDownloadWidget(self, self.competition)
urwid.connect_signal(self.task_widget, "download", self._on_download_task)
self.airspace_group: List[urwid.Widget] = []
self.airspace_pile = urwid.Pile(
[
self._make_file_radio(
sf,
self.airspace_group,
sf.name == self.competition.airspace,
self._on_airspace_changed,
)
for sf in self.airspaces
]
)
self.waypoint_group: List[urwid.Widget] = []
self.waypoint_pile = urwid.Pile(
[
self._make_file_radio(
sf,
self.waypoint_group,
sf.name == self.competition.waypoints,
self._on_waypoint_changed,
)
for sf in self.waypoints
]
)
self.profile_pile = urwid.Pile(
[self._make_profile_checkbox(prf) for prf in self.profiles]
)
self.download_status = urwid.Text("")
form = urwid.Pile(
[
self.class_widget,
urwid.Divider(),
self.task_widget,
urwid.Divider(),
self.download_status,
urwid.Divider(),
urwid.Text("Airspace files"),
p2(self.airspace_pile),
urwid.Divider(),
urwid.Text("Waypoint files"),
p2(self.waypoint_pile),
urwid.Divider(),
urwid.Text("XCSoar profiles"),
p2(self.profile_pile),
urwid.Divider(),
self._create_buttons(),
urwid.Divider(),
self._create_credits(),
]
)
filler = urwid.Filler(form, valign=urwid.TOP)
self.status = widget.CMFlashMessage(self)
return urwid.Frame(
widget.CMScreenPadding(filler),
header=widget.CMScreenHeader(self.competition.title),
footer=widget.CMScreenPadding(self.status),
)
def _create_buttons(self):
activate_btn = widget.CMButton("Activate")
urwid.connect_signal(activate_btn, "click", self._on_activate)
remove_btn = widget.CMButton("Remove")
# urwid.connect_signal(remove_btn, "click", self._on_remove)
self.connect_async(remove_btn, "click", self._on_remove)
return widget.ButtonRow([activate_btn, remove_btn])
def _create_credits(self) -> urwid.Widget:
credits = (
"Competiton files are provided by soaringspot.com. "
"XCSoar task is generated by soarscore.com."
)
return urwid.Text([("remark", credits)])
def _on_airspace_changed(self, ev, new_state, selected):
if not new_state:
return
self.competition.airspace = selected
storage.save_competition(self.competition)
self._udpate_xcsoar_config(self.competition)
self.status.flash(f"Airspace changed to: {selected}")
def _on_waypoint_changed(self, ev, new_state, selected):
if not new_state:
return
self.competition.waypoints = selected
storage.save_competition(self.competition)
self._udpate_xcsoar_config(self.competition)
self.status.flash(f"Waypoint changed to: {selected}")
def _on_profile_changed(self, ev, selected, profile: str) -> None:
if selected:
self.competition.add_profile(profile)
else:
self.competition.remove_profile(profile)
storage.save_competition(self.competition)
def _on_download_task(self, ev, task):
self.async_task(self._download_task(task))
def _on_class_changed(self, ev, new_class):
self.task_widget.refresh()
async def _download_task(self, taskinfo: soarscore.SoarScoreTaskInfo) -> None:
self.status.set(("progress", f"Downloading {taskinfo.title}..."))
task = await soarscore.fetch_url(taskinfo.task_url)
taskfname = xcsoar.install_default_task(task)
self.status.flash(
("success message", f"Task downloaded and installed: {taskfname}")
)
def _on_activate(self, btn) -> None:
if not self.competition.profiles:
self.status.flash(
("error message", "Please select at least one XCSoar profile")
)
return
profiles = ", ".join(self.competition.profiles)
self._udpate_xcsoar_config(self.competition)
self.status.flash(("success message", f"XCSoar profiles updated: {profiles}"))
async def _on_remove(self) -> None:
screen = CompetitionRemoveConfirmationScreen(self.container, self.competition)
screen.show()
confirmed = await screen.response
if confirmed:
storage.delete_competition(self.competition.id)
storage.get_settings().current_competition_id = None
storage.save_settings()
self.finish(None)
def _udpate_xcsoar_config(self, comp: storage.StoredCompetition) -> None:
for prf in self.competition.profiles:
xcprofile = xcsoar.get_xcsoar_profile(prf)
if comp.airspace:
xcprofile.set_airspace(
storage.get_full_file_path(comp.id, comp.airspace)
)
if comp.waypoints:
xcprofile.set_waypoint(
storage.get_full_file_path(comp.id, comp.waypoints)
)
xcprofile.save()
async def _update_competition_files(self) -> None:
compurl = self.competition.soaringspot_url
if compurl is None:
return
self.download_status.set_text(("progress", "Refreshing file list..."))
try:
downloads = await soaringspot.fetch_downloads(compurl)
except soaringspot.SoaringSpotClientError as e:
self.download_status.set_text(
("error message", f"Error refreshing file list: {e}")
)
log.exception("Error refreshing soaringspot downloads")
return
new_airspaces, new_waypoints = self._detect_new_files(downloads)
if not new_airspaces and not new_waypoints:
self.download_status.set_text(("remark", "No updates to competition files"))
return
self.download_status.set_text(
("success message", "New contest files detected!")
)
tasks = []
tasks.extend(
self._download_new_files(
new_airspaces,
self.airspace_pile,
self.airspace_group,
self._on_airspace_changed,
)
)
tasks.extend(
self._download_new_files(
new_waypoints,
self.waypoint_pile,
self.waypoint_group,
self._on_waypoint_changed,
)
)
await asyncio.gather(*tasks)
def _detect_new_files(self, downloads):
DFT = soaringspot.DownloadableFileType
airspace_files = {f.name for f in self.airspaces}
dl_airspaces = {d.filename for d in downloads if d.kind == DFT.AIRSPACE}
new_airspaces = dl_airspaces - airspace_files
waypoint_files = {f.name for f in self.waypoints}
dl_waypoints = {d.filename for d in downloads if d.kind == DFT.WAYPOINT}
new_waypoints = dl_waypoints - waypoint_files
dl_idx = {d.filename: d for d in downloads}
return ([dl_idx[f] for f in new_airspaces], [dl_idx[f] for f in new_waypoints])
def _download_new_files(self, new_files, checkbox_pile, group, select_handler):
tasks = []
for sspotfile in new_files:
sf = storage.StoredFile(name=sspotfile.filename, size=None)
radio = self._make_file_radio(sf, group, False, select_handler)
checkbox_pile.contents.insert(0, (radio, ("pack", None)))
tasks.append(self._download_file(sf, sspotfile.href, radio))
return tasks
async def _download_file(
self, sf: storage.StoredFile, url: str, radio: urwid.RadioButton
) -> None:
orig_radio = radio.original_widget
orig_radio.set_label(self._make_label(sf, [("progress", "Downloading...")]))
dlcontents = await http.fetch_file(url)
stored = storage.store_file(self.competition.id, sf.name, dlcontents)
orig_radio.set_label(self._make_label(stored, [("success banner", " New! ")]))
def _make_file_radio(
self, sf: storage.StoredFile, group, selected: bool, select_handler
) -> urwid.RadioButton:
label = f"{sf.name} ({sf.format_size()})"
radio = urwid.RadioButton(group, label, selected)
urwid.connect_signal(radio, "change", select_handler, sf.name)
return urwid.AttrMap(radio, "li normal", "li focus")
def _make_label(
self, sf: storage.StoredFile, extra: widget.UrwidMarkup = None
) -> widget.UrwidMarkup:
label = f"{sf.name} ({sf.format_size()})"
markup: List[widget.UrwidMarkup] = [label, " "]
markup.extend(extra or [])
return cast(widget.UrwidMarkup, markup)
def _make_profile_checkbox(self, profile: str) -> urwid.CheckBox:
cb = urwid.CheckBox(profile, profile in self.competition.profiles)
urwid.connect_signal(cb, "change", self._on_profile_changed, profile)
return urwid.AttrMap(cb, "li normal", "li focus")