72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import rofi
|
||
|
||
from entity.response import SearchResponse
|
||
from helper.api_helper import run_search, run_delete
|
||
from helper.file_helper import edit, edit_and_copy
|
||
|
||
|
||
class SearchMenu:
|
||
def __init__(self, r: rofi.Rofi):
|
||
self._r = r
|
||
|
||
def run(self):
|
||
while True:
|
||
key = self._r.text_entry("搜索", allow_blank=True)
|
||
if key is None:
|
||
break
|
||
print(key)
|
||
try:
|
||
res = run_search(key)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
self._r.error(str(e))
|
||
return
|
||
if res.status != "SUCCESS":
|
||
self._r.error(res.data)
|
||
return
|
||
SearchResultMenu(self._r, res.data).run()
|
||
|
||
|
||
class SearchResultMenu:
|
||
def __init__(self, r: rofi.Rofi, data: list[SearchResponse.SearchData]):
|
||
self._r = r
|
||
self._data = data
|
||
|
||
def run(self):
|
||
while True:
|
||
if len(self._data) == 0:
|
||
self._r.select("搜索结果", ["未找到Snippet记录"])
|
||
break
|
||
else:
|
||
# 将data按照索引和path换成dict，前者为键
|
||
options = [f"{d.name} - {d.path}" for d in self._data]
|
||
index, key = self._r.select("搜索结果", options)
|
||
if key == -1:
|
||
break
|
||
SearchResultActionMenu(self._r, self._data[index]).run()
|
||
|
||
|
||
class SearchResultActionMenu:
|
||
def __init__(self, r: rofi.Rofi, data: SearchResponse.SearchData):
|
||
self._r = r
|
||
self.data = data
|
||
|
||
def run(self):
|
||
while True:
|
||
index, key = self._r.select("操作", ["编辑", "编辑并复制", "删除"])
|
||
if key == -1:
|
||
break
|
||
match index:
|
||
case 0:
|
||
result = edit(self.data.path, self.data.id)
|
||
if not result.ok:
|
||
self._r.error(result.message)
|
||
case 1:
|
||
result = edit_and_copy(self.data.path, self.data.id)
|
||
if not result.ok:
|
||
self._r.error(result.message)
|
||
exit()
|
||
case 2:
|
||
run_delete(self.data.path)
|
||
break
|