From 957d749f1c9bf6d82a2fdbb62776e275b84ea398 Mon Sep 17 00:00:00 2001
From: Mumtaz Hajjo Alrifai <mumtazrifai@protonmail.com>
Date: Sat, 15 Feb 2025 23:15:11 -0500
Subject: [PATCH] Add function to restore the default name of a flag

---
 qt/aqt/flags.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/qt/aqt/flags.py b/qt/aqt/flags.py
index 7a1b9aef1..032382cad 100644
--- a/qt/aqt/flags.py
+++ b/qt/aqt/flags.py
@@ -35,6 +35,15 @@ class FlagManager:
     def __init__(self, mw: aqt.main.AnkiQt) -> None:
         self.mw = mw
         self._flags: list[Flag] = []
+        self.default_flag_names: dict[int, str] = {
+            1: tr.actions_flag_red(),
+            2: tr.actions_flag_orange(),
+            3: tr.actions_flag_green(),
+            4: tr.actions_flag_blue(),
+            5: tr.actions_flag_pink(),
+            6: tr.actions_flag_turquoise(),
+            7: tr.actions_flag_purple()
+        }
 
     def all(self) -> list[Flag]:
         """Return a list of all flags."""
@@ -55,6 +64,15 @@ class FlagManager:
         self.mw.col.set_config("flagLabels", labels)
         gui_hooks.flag_label_did_change()
 
+    def restore_default_flag_name(self, flag_index: int) -> None:
+        labels = self.mw.col.get_config("flagLabels", {})
+        if str(flag_index) not in labels:
+            return
+        del labels[str(flag_index)]
+        self.get_flag(flag_index).label = self.default_flag_names[flag_index]
+        self.mw.col.set_config("flagLabels", labels)
+        gui_hooks.flag_label_did_change()
+
     def require_refresh(self) -> None:
         "Discard cached labels."
         self._flags = []