add src_download_link to ProjImgTrans (eb8d000a) · Commits · git-mirror / BallonsTranslator

ballontranslator/ui/imgtrans_proj.py

+10 −1

Original line number	Diff line number	Diff line
		@@ -36,6 +36,9 @@ class ProjImgTrans:
		self.not_found_pages: Dict[List[TextBlock]] = {}
		self.new_pages: List[str] = []
		self.proj_path: str = None

		self.src_download_link: str = ''

		self.current_img: str = None
		self.img_array: np.ndarray = None
		self.mask_array: np.ndarray = None
		@@ -86,6 +89,9 @@ class ProjImgTrans:
		def result_dir(self):
		return osp.join(self.directory, 'result')

		def merge_variables(self, src_download_link: str = '', **kwargs):
		self.src_download_link = src_download_link

		def load_from_dict(self, proj_dict: dict):
		self.set_current_img(None)
		try:
		@@ -107,6 +113,7 @@ class ProjImgTrans:
		self._idx2pagename[ii] = imname
		for imname in not_found_pages:
		self.not_found_pages[imname] = [TextBlock(**blk_dict) for blk_dict in page_dict[imname]]
		self.merge_variables(**proj_dict)
		except Exception as e:
		raise ProjectNotSupportedException(e)
		set_img_failed = False
		@@ -185,6 +192,7 @@ class ProjImgTrans:
		self.pages[imgname] = []
		self._pagename2idx[imgname] = ii
		self._idx2pagename[ii] = imgname
		self.merge_variables()
		self.set_current_img_byidx(0)
		self.save()

		@@ -200,7 +208,8 @@ class ProjImgTrans:
		return {
		'directory': self.directory,
		'pages': pages,
		'current_img': self.current_img
		'current_img': self.current_img,
		'src_download_link': self.src_download_link
		}

		def read_img(self, imgname: str) -> np.ndarray: