Exemple #1
0
	def dropEvent(self, event):
		acceptable = []
		unaccept = []
		for u in event.mimeData().urls():
			path = u.toLocalFile()
			if os.path.isdir(path) or path.endswith(utils.ARCHIVE_FILES):
				acceptable.append(path)
			else:
				unaccept.append(path)
		log_i('Acceptable dropped items: {}'.format(len(acceptable)))
		log_i('Unacceptable dropped items: {}'.format(len(unaccept)))
		log_d('Dropped items: {}\n{}'.format(acceptable, unaccept).encode(errors='ignore'))
		if acceptable:
			self.notification_bar.add_text('Adding dropped items...')
			log_i('Adding dropped items')
			l = len(acceptable) == 1
			f_item = acceptable[0]
			if f_item.endswith(utils.ARCHIVE_FILES):
				f_item = utils.check_archive(f_item)
			else:
				f_item = utils.recursive_gallery_check(f_item)
			f_item_l = len(f_item) < 2
			subfolder_as_c = not app_constants.SUBFOLDER_AS_GALLERY
			if l and subfolder_as_c or l and f_item_l:
				g_d = gallerydialog.GalleryDialog(self, acceptable[0])
				g_d.SERIES.connect(self.manga_list_view.gallery_model.addRows)
				g_d.show()
			else:
				self.gallery_populate(acceptable, True)
		else:
			text = 'File not supported' if len(unaccept) < 2 else 'Files not supported'
			self.notification_bar.add_text(text)

		if unaccept:
			self.notification_bar.add_text('Some unsupported files did not get added')
Exemple #2
0
 def choose_dir(self, mode):
     """
     Pass which mode to open the folder explorer in:
     'f': directory
     'a': files
     Or pass a predefined path
     """
     self.done.show()
     self.file_exists_lbl.hide()
     if mode == 'a':
         name = QFileDialog.getOpenFileName(self,
                                            'Choose archive',
                                            filter=utils.FILE_FILTER)
         name = name[0]
     elif mode == 'f':
         name = QFileDialog.getExistingDirectory(self, 'Choose folder')
     elif mode:
         if os.path.exists(mode):
             name = mode
         else:
             return None
     if not name:
         return
     head, tail = os.path.split(name)
     name = os.path.join(head, tail)
     parsed = utils.title_parser(tail)
     self.title_edit.setText(parsed['title'])
     self.author_edit.setText(parsed['artist'])
     self.path_lbl.setText(name)
     if not parsed['language']:
         parsed['language'] = app_constants.G_DEF_LANGUAGE
     l_i = self.lang_box.findText(parsed['language'])
     if l_i != -1:
         self.lang_box.setCurrentIndex(l_i)
     if gallerydb.GalleryDB.check_exists(name):
         self.file_exists_lbl.setText(
             '<font color="red">Gallery already exists.</font>')
         self.file_exists_lbl.show()
     # check galleries
     gs = 1
     if name.endswith(utils.ARCHIVE_FILES):
         gs = len(utils.check_archive(name))
     elif os.path.isdir(name):
         g_dirs, g_archs = utils.recursive_gallery_check(name)
         gs = len(g_dirs) + len(g_archs)
     if gs == 0:
         self.file_exists_lbl.setText(
             '<font color="red">Invalid gallery source.</font>')
         self.file_exists_lbl.show()
         self.done.hide()
     if app_constants.SUBFOLDER_AS_GALLERY:
         if gs > 1:
             self.file_exists_lbl.setText(
                 '<font color="red">More than one galleries detected in source! Use other methods to add.</font>'
             )
             self.file_exists_lbl.show()
             self.done.hide()
	def choose_dir(self, mode):
		"""
		Pass which mode to open the folder explorer in:
		'f': directory
		'a': files
		Or pass a predefined path
		"""
		self.done.show()
		self.file_exists_lbl.hide()
		if mode == 'a':
			name = QFileDialog.getOpenFileName(self, 'Choose archive',
											  filter=utils.FILE_FILTER)
			name = name[0]
		elif mode == 'f':
			name = QFileDialog.getExistingDirectory(self, 'Choose folder')
		elif mode:
			if os.path.exists(mode):
				name = mode
			else:
				return None
		if not name:
			return
		head, tail = os.path.split(name)
		name = os.path.join(head, tail)
		parsed = utils.title_parser(tail)
		self.title_edit.setText(parsed['title'])
		self.author_edit.setText(parsed['artist'])
		self.path_lbl.setText(name)
		if not parsed['language']:
			parsed['language'] = app_constants.G_DEF_LANGUAGE
		l_i = self.lang_box.findText(parsed['language'])
		if l_i != -1:
			self.lang_box.setCurrentIndex(l_i)
		if gallerydb.GalleryDB.check_exists(name):
			self.file_exists_lbl.setText('<font color="red">Gallery already exists.</font>')
			self.file_exists_lbl.show()
		# check galleries
		gs = 1
		if name.endswith(utils.ARCHIVE_FILES):
			gs = len(utils.check_archive(name))
		elif os.path.isdir(name):
			g_dirs, g_archs = utils.recursive_gallery_check(name)
			gs = len(g_dirs) + len(g_archs)
		if gs == 0:
			self.file_exists_lbl.setText('<font color="red">Invalid gallery source.</font>')
			self.file_exists_lbl.show()
			self.done.hide()
		if app_constants.SUBFOLDER_AS_GALLERY:
			if gs > 1:
				self.file_exists_lbl.setText('<font color="red">More than one galleries detected in source! Use other methods to add.</font>')
				self.file_exists_lbl.show()
				self.done.hide()
Exemple #4
0
 def on_created(self, event):
     if not app_constants.OVERRIDE_MONITOR:
         if self.file_filter(event):
             gs = 0
             if event.src_path.endswith(utils.ARCHIVE_FILES):
                 gs = len(utils.check_archive(event.src_path))
             elif event.is_directory:
                 g_dirs, g_archs = utils.recursive_gallery_check(event.src_path)
                 gs = len(g_dirs) + len(g_archs)
             if gs:
                 self.CREATE_SIGNAL.emit(event.src_path)
     else:
         app_constants.OVERRIDE_MONITOR = False
Exemple #5
0
 def on_created(self, event):
     if not app_constants.OVERRIDE_MONITOR:
         if self.file_filter(event):
             gs = 0
             if event.src_path.endswith(utils.ARCHIVE_FILES):
                 gs = len(utils.check_archive(event.src_path))
             elif event.is_directory:
                 g_dirs, g_archs = utils.recursive_gallery_check(
                     event.src_path)
                 gs = len(g_dirs) + len(g_archs)
             if gs:
                 self.CREATE_SIGNAL.emit(event.src_path)
     else:
         app_constants.OVERRIDE_MONITOR = False
Exemple #6
0
    def set_chapters(self, gallery_object, add_to_model=True):
        path = gallery_object.path
        chap_container = gallerydb.ChaptersContainer(gallery_object)
        metafile = utils.GMetafile()
        try:
            log_d('Listing dir...')
            con = scandir.scandir(path)  # list all folders in gallery dir
            log_i('Gallery source is a directory')
            log_d('Sorting')
            chapters = sorted([
                sub.path for sub in con
                if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)
            ])  #subfolders
            # if gallery has chapters divided into sub folders
            if len(chapters) != 0:
                log_d('Chapters divided in folders..')
                for ch in chapters:
                    chap = chap_container.create_chapter()
                    chap.title = utils.title_parser(ch)['title']
                    chap.path = os.path.join(path, ch)
                    metafile.update(utils.GMetafile(chap.path))
                    chap.pages = len(list(scandir.scandir(chap.path)))

            else:  #else assume that all images are in gallery folder
                chap = chap_container.create_chapter()
                chap.title = utils.title_parser(
                    os.path.split(path)[1])['title']
                chap.path = path
                metafile.update(utils.GMetafile(path))
                chap.pages = len(list(scandir.scandir(path)))

        except NotADirectoryError:
            if path.endswith(utils.ARCHIVE_FILES):
                gallery_object.is_archive = 1
                log_i("Gallery source is an archive")
                archive_g = sorted(utils.check_archive(path))
                for g in archive_g:
                    chap = chap_container.create_chapter()
                    chap.path = g
                    chap.in_archive = 1
                    metafile.update(utils.GMetafile(g, path))
                    arch = utils.ArchiveFile(path)
                    chap.pages = len(arch.dir_contents(g))
                    arch.close()

        metafile.apply_gallery(gallery_object)
        if add_to_model:
            self.SERIES.emit([gallery_object])
            log_d('Sent gallery to model')
	def set_chapters(self, gallery_object, add_to_model=True):
		path = gallery_object.path
		chap_container = gallerydb.ChaptersContainer(gallery_object)
		metafile = utils.GMetafile()
		try:
			log_d('Listing dir...')
			con = scandir.scandir(path) # list all folders in gallery dir
			log_i('Gallery source is a directory')
			log_d('Sorting')
			chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)]) #subfolders
			# if gallery has chapters divided into sub folders
			if len(chapters) != 0:
				log_d('Chapters divided in folders..')
				for ch in chapters:
					chap = chap_container.create_chapter()
					chap.title = utils.title_parser(ch)['title']
					chap.path = os.path.join(path, ch)
					metafile.update(utils.GMetafile(chap.path))
					chap.pages = len(list(scandir.scandir(chap.path)))

			else: #else assume that all images are in gallery folder
				chap = chap_container.create_chapter()
				chap.title = utils.title_parser(os.path.split(path)[1])['title']
				chap.path = path
				metafile.update(utils.GMetafile(path))
				chap.pages = len(list(scandir.scandir(path)))

		except NotADirectoryError:
			if path.endswith(utils.ARCHIVE_FILES):
				gallery_object.is_archive = 1
				log_i("Gallery source is an archive")
				archive_g = sorted(utils.check_archive(path))
				for g in archive_g:
					chap = chap_container.create_chapter()
					chap.path = g
					chap.in_archive = 1
					metafile.update(utils.GMetafile(g, path))
					arch = utils.ArchiveFile(path)
					chap.pages = len(arch.dir_contents(g))
					arch.close()

		metafile.apply_gallery(gallery_object)
		if add_to_model:
			self.SERIES.emit([gallery_object])
			log_d('Sent gallery to model')
    def set_chapters(self, gallery_object, add_to_model=True):
        path = gallery_object.path
        try:
            log_d("Listing dir...")
            con = scandir.scandir(path)  # list all folders in gallery dir
            log_i("Gallery source is a directory")
            log_d("Sorting")
            chapters = sorted(
                [sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)]
            )  # subfolders
            # if gallery has chapters divided into sub folders
            if len(chapters) != 0:
                log_d("Chapters divided in folders..")
                for numb, ch in enumerate(chapters):
                    chap_path = os.path.join(path, ch)
                    gallery_object.chapters[numb] = chap_path

            else:  # else assume that all images are in gallery folder
                gallery_object.chapters[0] = path

                # find last edited file
            times = set()
            log_d("Finding last update...")
            for root, dirs, files in scandir.walk(path, topdown=False):
                for img in files:
                    fp = os.path.join(root, img)
                    times.add(os.path.getmtime(fp))
            gallery_object.last_update = time.asctime(time.gmtime(max(times)))
            log_d("Found last update")
        except NotADirectoryError:
            if path.endswith(utils.ARCHIVE_FILES):
                gallery_object.is_archive = 1
                log_i("Gallery source is an archive")
                archive_g = sorted(utils.check_archive(path))
                for n, g in enumerate(archive_g):
                    gallery_object.chapters[n] = g

        if add_to_model:
            self.SERIES.emit([gallery_object])
            log_d("Sent gallery to model")
Exemple #9
0
    def create_gallery(self,
                       path,
                       folder_name,
                       do_chapters=True,
                       archive=None):
        is_archive = True if archive else False
        temp_p = archive if is_archive else path
        folder_name = folder_name or path if folder_name or path else os.path.split(
            archive)[1]
        if utils.check_ignore_list(temp_p) and not GalleryDB.check_exists(
                temp_p, self.galleries_from_db, False):
            log_i('Creating gallery: {}'.format(
                folder_name.encode('utf-8', 'ignore')))
            new_gallery = Gallery()
            images_paths = []
            metafile = utils.GMetafile()
            try:
                con = scandir.scandir(
                    temp_p)  #all of content in the gallery folder
                log_i('Gallery source is a directory')
                chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
                    if do_chapters else [] #subfolders
                # if gallery has chapters divided into sub folders
                numb_of_chapters = len(chapters)
                if numb_of_chapters != 0:
                    log_i('Gallery has {} chapters'.format(numb_of_chapters))
                    for ch in chapters:
                        chap = new_gallery.chapters.create_chapter()
                        chap.title = utils.title_parser(ch)['title']
                        chap.path = os.path.join(path, ch)
                        chap.pages = len([
                            x for x in scandir.scandir(chap.path)
                            if x.name.endswith(utils.IMG_FILES)
                        ])
                        metafile.update(utils.GMetafile(chap.path))

                else:  #else assume that all images are in gallery folder
                    chap = new_gallery.chapters.create_chapter()
                    chap.title = utils.title_parser(
                        os.path.split(path)[1])['title']
                    chap.path = path
                    metafile.update(utils.GMetafile(chap.path))
                    chap.pages = len(list(scandir.scandir(path)))

                parsed = utils.title_parser(folder_name)
            except NotADirectoryError:
                try:
                    if is_archive or temp_p.endswith(utils.ARCHIVE_FILES):
                        log_i('Gallery source is an archive')
                        contents = utils.check_archive(temp_p)
                        if contents:
                            new_gallery.is_archive = 1
                            new_gallery.path_in_archive = '' if not is_archive else path
                            if folder_name.endswith('/'):
                                folder_name = folder_name[:-1]
                                fn = os.path.split(folder_name)
                                folder_name = fn[1] or fn[2]
                            folder_name = folder_name.replace('/', '')
                            if folder_name.endswith(utils.ARCHIVE_FILES):
                                n = folder_name
                                for ext in utils.ARCHIVE_FILES:
                                    n = n.replace(ext, '')
                                parsed = utils.title_parser(n)
                            else:
                                parsed = utils.title_parser(folder_name)

                            if do_chapters:
                                archive_g = sorted(contents)
                                if not archive_g:
                                    log_w('No chapters found for {}'.format(
                                        temp_p.encode(errors='ignore')))
                                    raise ValueError
                                for g in archive_g:
                                    chap = new_gallery.chapters.create_chapter(
                                    )
                                    chap.in_archive = 1
                                    chap.title = parsed[
                                        'title'] if not g else utils.title_parser(
                                            g.replace('/', ''))['title']
                                    chap.path = g
                                    metafile.update(utils.GMetafile(g, temp_p))
                                    arch = utils.ArchiveFile(temp_p)
                                    chap.pages = len([
                                        x for x in arch.dir_contents(g)
                                        if x.endswith(utils.IMG_FILES)
                                    ])
                                    arch.close()
                            else:
                                chap = new_gallery.chapters.create_chapter()
                                chap.title = utils.title_parser(
                                    os.path.split(path)[1])['title']
                                chap.in_archive = 1
                                chap.path = path
                                metafile.update(utils.GMetafile(path, temp_p))
                                arch = utils.ArchiveFile(temp_p)
                                chap.pages = len(arch.dir_contents(''))
                                arch.close()
                        else:
                            raise ValueError
                    else:
                        raise ValueError
                except ValueError:
                    log_w('Skipped {} in local search'.format(
                        path.encode(errors='ignore')))
                    self.skipped_paths.append((
                        temp_p,
                        'Empty archive',
                    ))
                    return
                except app_constants.CreateArchiveFail:
                    log_w('Skipped {} in local search'.format(
                        path.encode(errors='ignore')))
                    self.skipped_paths.append((
                        temp_p,
                        'Error creating archive',
                    ))
                    return
                except app_constants.TitleParsingError:
                    log_w('Skipped {} in local search'.format(
                        path.encode(errors='ignore')))
                    self.skipped_paths.append((
                        temp_p,
                        'Error while parsing folder/archive name',
                    ))
                    return

            new_gallery.title = parsed['title']
            new_gallery.path = temp_p
            new_gallery.artist = parsed['artist']
            new_gallery.language = parsed['language']
            new_gallery.info = ""
            new_gallery.view = app_constants.ViewType.Addition
            metafile.apply_gallery(new_gallery)

            if app_constants.MOVE_IMPORTED_GALLERIES and not app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
                new_gallery.move_gallery()

            self.LOCAL_EMITTER.emit(new_gallery)
            self._data.append(new_gallery)
            log_i('Gallery successful created: {}'.format(
                folder_name.encode('utf-8', 'ignore')))
            return True
        else:
            log_i('Gallery already exists or ignored: {}'.format(
                folder_name.encode('utf-8', 'ignore')))
            self.skipped_paths.append((temp_p, 'Already exists or ignored'))
            return False
Exemple #10
0
    def local(self, s_path=None):
        """
        Do a local search in the given series_path.
        """
        self._data.clear()
        if s_path:
            self.series_path = s_path
        try:
            gallery_l = sorted([
                p.name for p in scandir.scandir(self.series_path)
            ])  #list of folders in the "Gallery" folder
            mixed = False
        except TypeError:
            gallery_l = self.series_path
            mixed = True
        if len(gallery_l) != 0:  # if gallery path list is not empty
            log_i('Gallery folder is not empty')
            if len(self.galleries_from_db) != len(app_constants.GALLERY_DATA):
                self._refresh_filter_list()
            self.DATA_COUNT.emit(len(
                gallery_l))  #tell model how many items are going to be added
            log_i('Received {} paths'.format(len(gallery_l)))
            progress = 0

            for folder_name in gallery_l:  # folder_name = gallery folder title
                self._curr_gallery = folder_name
                if mixed:
                    path = folder_name
                    folder_name = os.path.split(path)[1]
                else:
                    path = os.path.join(self.series_path, folder_name)
                if app_constants.SUBFOLDER_AS_GALLERY or app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                    if app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                        app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY = False
                    log_i("Treating each subfolder as gallery")
                    if os.path.isdir(path):
                        gallery_folders, gallery_archives = utils.recursive_gallery_check(
                            path)
                        for gs in gallery_folders:
                            self.create_gallery(gs,
                                                os.path.split(gs)[1], False)
                        p_saving = {}
                        for gs in gallery_archives:

                            self.create_gallery(gs[0],
                                                os.path.split(gs[0])[1],
                                                False,
                                                archive=gs[1])
                    elif path.endswith(utils.ARCHIVE_FILES):
                        for g in utils.check_archive(path):
                            self.create_gallery(g,
                                                os.path.split(g)[1],
                                                False,
                                                archive=path)
                else:
                    try:
                        if os.path.isdir(path):
                            if not list(scandir.scandir(path)):
                                raise ValueError
                        elif not path.endswith(utils.ARCHIVE_FILES):
                            raise NotADirectoryError

                        log_i("Treating each subfolder as chapter")
                        self.create_gallery(path,
                                            folder_name,
                                            do_chapters=True)

                    except ValueError:
                        self.skipped_paths.append((path, 'Empty directory'))
                        log_w('Directory is empty: {}'.format(
                            path.encode(errors='ignore')))
                    except NotADirectoryError:
                        self.skipped_paths.append((path, 'Unsupported file'))
                        log_w('Unsupported file: {}'.format(
                            path.encode(errors='ignore')))

                progress += 1  # update the progress bar
                self.PROGRESS.emit(progress)
        else:  # if gallery folder is empty
            log_e('Local search error: Invalid directory')
            log_e('Gallery folder is empty')
            app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True  # sanity check
            self.FINISHED.emit(False)
            # might want to include an error message
        app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = False
        # everything went well
        log_i('Local search: OK')
        log_i('Created {} items'.format(len(self._data)))
        if self._to_queue_container:
            for x in self._data:
                self._galleries_queue.put(x)
        else:
            self.FINISHED.emit(self._data)
        if self.skipped_paths:
            self.SKIPPED.emit(self.skipped_paths)
Exemple #11
0
	def local(self, s_path=None):
		"""
		Do a local search in the given series_path.
		"""
		self.data.clear()
		if s_path:
			self.series_path = s_path
		try:
			gallery_l = sorted([p.name for p in scandir.scandir(self.series_path)]) #list of folders in the "Gallery" folder
			mixed = False
		except TypeError:
			gallery_l = self.series_path
			mixed = True
		if len(gallery_l) != 0: # if gallery path list is not empty
			log_i('Gallery folder is not empty')
			try:
				if len(self.galleries_from_db) != len(app_constants.GALLERY_DATA):
					self._refresh_filter_list()
				self.DATA_COUNT.emit(len(gallery_l)) #tell model how many items are going to be added
				log_i('Received {} paths'.format(len(gallery_l)))
				progress = 0
				def create_gallery(path, folder_name, do_chapters=True, archive=None):
					is_archive = True if archive else False
					temp_p = archive if is_archive else path
					folder_name = folder_name or path if folder_name or path else os.path.split(archive)[1]
					if utils.check_ignore_list(temp_p) and not GalleryDB.check_exists(temp_p, self.galleries_from_db, False):
						log_i('Creating gallery: {}'.format(folder_name.encode('utf-8', 'ignore')))
						new_gallery = Gallery()
						images_paths = []
						metafile = utils.GMetafile()
						try:
							con = scandir.scandir(temp_p) #all of content in the gallery folder
							log_i('Gallery source is a directory')
							chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
							    if do_chapters else [] #subfolders
							# if gallery has chapters divided into sub folders
							numb_of_chapters = len(chapters)
							if numb_of_chapters != 0:
								log_i('Gallery has {} chapters'.format(numb_of_chapters))
								for ch in chapters:
									chap = new_gallery.chapters.create_chapter()
									chap.title = utils.title_parser(ch)['title']
									chap.path = os.path.join(path, ch)
									chap.pages = len(list(scandir.scandir(chap.path)))
									metafile.update(utils.GMetafile(chap.path))

							else: #else assume that all images are in gallery folder
								chap = new_gallery.chapters.create_chapter()
								chap.title = utils.title_parser(os.path.split(path)[1])['title']
								chap.path = path
								metafile.update(utils.GMetafile(chap.path))
								chap.pages = len(list(scandir.scandir(path)))
				
							parsed = utils.title_parser(folder_name)
						except NotADirectoryError:
							try:
								if is_archive or temp_p.endswith(utils.ARCHIVE_FILES):
									log_i('Gallery source is an archive')
									contents = utils.check_archive(temp_p)
									if contents:
										new_gallery.is_archive = 1
										new_gallery.path_in_archive = '' if not is_archive else path
										if folder_name.endswith('/'):
											folder_name = folder_name[:-1]
											fn = os.path.split(folder_name)
											folder_name = fn[1] or fn[2]
										folder_name = folder_name.replace('/','')
										if folder_name.endswith(utils.ARCHIVE_FILES):
											n = folder_name
											for ext in utils.ARCHIVE_FILES:
												n = n.replace(ext, '')
											parsed = utils.title_parser(n)
										else:
											parsed = utils.title_parser(folder_name)
												
										if do_chapters:
											archive_g = sorted(contents)
											if not archive_g:
												log_w('No chapters found for {}'.format(temp_p.encode(errors='ignore')))
												raise ValueError
											for g in archive_g:
												chap = new_gallery.chapters.create_chapter()
												chap.in_archive = 1
												chap.title = utils.title_parser(g)['title']
												chap.path = g
												metafile.update(utils.GMetafile(g, temp_p))
												arch = utils.ArchiveFile(temp_p)
												chap.pages = len(arch.dir_contents(g))
												arch.close()
										else:
											chap = new_gallery.chapters.create_chapter()
											chap.title = utils.title_parser(os.path.split(path)[1])['title']
											chap.in_archive = 1
											chap.path = path
											metafile.update(utils.GMetafile(path, temp_p))
											arch = utils.ArchiveFile(temp_p)
											chap.pages = len(arch.dir_contents(''))
											arch.close()
									else:
										raise ValueError
								else:
									raise ValueError
							except ValueError:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append((temp_p, 'Empty archive',))
								return
							except app_constants.CreateArchiveFail:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append((temp_p, 'Error creating archive',))
								return

						new_gallery.title = parsed['title']
						new_gallery.path = temp_p
						new_gallery.artist = parsed['artist']
						new_gallery.language = parsed['language']
						new_gallery.info = ""
						metafile.apply_gallery(new_gallery)
						if app_constants.MOVE_IMPORTED_GALLERIES and not app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
							new_gallery.path = utils.move_files(temp_p)

						self.data.append(new_gallery)
						log_i('Gallery successful created: {}'.format(folder_name.encode('utf-8', 'ignore')))
					else:
						log_i('Gallery already exists: {}'.format(folder_name.encode('utf-8', 'ignore')))
						self.skipped_paths.append((temp_p, 'Already exists'))

				for folder_name in gallery_l: # folder_name = gallery folder title
					self._curr_gallery = folder_name
					if mixed:
						path = folder_name
						folder_name = os.path.split(path)[1]
					else:
						path = os.path.join(self.series_path, folder_name)
					if app_constants.SUBFOLDER_AS_GALLERY or app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
						if app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
							app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY = False
						log_i("Treating each subfolder as gallery")
						if os.path.isdir(path):
							gallery_folders, gallery_archives = utils.recursive_gallery_check(path)
							for gs in gallery_folders:
									create_gallery(gs, os.path.split(gs)[1], False)
							p_saving = {}
							for gs in gallery_archives:
									
									create_gallery(gs[0], os.path.split(gs[0])[1], False, archive=gs[1])
						elif path.endswith(utils.ARCHIVE_FILES):
							for g in utils.check_archive(path):
								create_gallery(g, os.path.split(g)[1], False, archive=path)
					else:
						try:
							if os.path.isdir(path):
								if not list(scandir.scandir(path)):
									raise ValueError
							elif not path.endswith(utils.ARCHIVE_FILES):
								raise NotADirectoryError

							log_i("Treating each subfolder as chapter")
							create_gallery(path, folder_name, do_chapters=True)

						except ValueError:
							self.skipped_paths.append((path, 'Empty directory'))
							log_w('Directory is empty: {}'.format(path.encode(errors='ignore')))
						except NotADirectoryError:
							self.skipped_paths.append((path, 'Unsupported file'))
							log_w('Unsupported file: {}'.format(path.encode(errors='ignore')))

					progress += 1 # update the progress bar
					self.PROGRESS.emit(progress)
			except:
				log.exception('Local Search Error:')
				app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True # sanity check
				self.FINISHED.emit(False)
		else: # if gallery folder is empty
			log_e('Local search error: Invalid directory')
			log_e('Gallery folder is empty')
			app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True # sanity check
			self.FINISHED.emit(False)
			# might want to include an error message
		app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = False
		# everything went well
		log_i('Local search: OK')
		log_i('Created {} items'.format(len(self.data)))
		self.FINISHED.emit(self.data)
		if self.skipped_paths:
			self.SKIPPED.emit(self.skipped_paths)
Exemple #12
0
				def create_gallery(path, folder_name, do_chapters=True, archive=None):
					is_archive = True if archive else False
					temp_p = archive if is_archive else path
					folder_name = folder_name or path if folder_name or path else os.path.split(archive)[1]
					if utils.check_ignore_list(temp_p) and not GalleryDB.check_exists(temp_p, self.galleries_from_db, False):
						log_i('Creating gallery: {}'.format(folder_name.encode('utf-8', 'ignore')))
						new_gallery = Gallery()
						images_paths = []
						metafile = utils.GMetafile()
						try:
							con = scandir.scandir(temp_p) #all of content in the gallery folder
							log_i('Gallery source is a directory')
							chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
							    if do_chapters else [] #subfolders
							# if gallery has chapters divided into sub folders
							numb_of_chapters = len(chapters)
							if numb_of_chapters != 0:
								log_i('Gallery has {} chapters'.format(numb_of_chapters))
								for ch in chapters:
									chap = new_gallery.chapters.create_chapter()
									chap.title = utils.title_parser(ch)['title']
									chap.path = os.path.join(path, ch)
									chap.pages = len(list(scandir.scandir(chap.path)))
									metafile.update(utils.GMetafile(chap.path))

							else: #else assume that all images are in gallery folder
								chap = new_gallery.chapters.create_chapter()
								chap.title = utils.title_parser(os.path.split(path)[1])['title']
								chap.path = path
								metafile.update(utils.GMetafile(chap.path))
								chap.pages = len(list(scandir.scandir(path)))
				
							parsed = utils.title_parser(folder_name)
						except NotADirectoryError:
							try:
								if is_archive or temp_p.endswith(utils.ARCHIVE_FILES):
									log_i('Gallery source is an archive')
									contents = utils.check_archive(temp_p)
									if contents:
										new_gallery.is_archive = 1
										new_gallery.path_in_archive = '' if not is_archive else path
										if folder_name.endswith('/'):
											folder_name = folder_name[:-1]
											fn = os.path.split(folder_name)
											folder_name = fn[1] or fn[2]
										folder_name = folder_name.replace('/','')
										if folder_name.endswith(utils.ARCHIVE_FILES):
											n = folder_name
											for ext in utils.ARCHIVE_FILES:
												n = n.replace(ext, '')
											parsed = utils.title_parser(n)
										else:
											parsed = utils.title_parser(folder_name)
												
										if do_chapters:
											archive_g = sorted(contents)
											if not archive_g:
												log_w('No chapters found for {}'.format(temp_p.encode(errors='ignore')))
												raise ValueError
											for g in archive_g:
												chap = new_gallery.chapters.create_chapter()
												chap.in_archive = 1
												chap.title = utils.title_parser(g)['title']
												chap.path = g
												metafile.update(utils.GMetafile(g, temp_p))
												arch = utils.ArchiveFile(temp_p)
												chap.pages = len(arch.dir_contents(g))
												arch.close()
										else:
											chap = new_gallery.chapters.create_chapter()
											chap.title = utils.title_parser(os.path.split(path)[1])['title']
											chap.in_archive = 1
											chap.path = path
											metafile.update(utils.GMetafile(path, temp_p))
											arch = utils.ArchiveFile(temp_p)
											chap.pages = len(arch.dir_contents(''))
											arch.close()
									else:
										raise ValueError
								else:
									raise ValueError
							except ValueError:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append((temp_p, 'Empty archive',))
								return
							except app_constants.CreateArchiveFail:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append((temp_p, 'Error creating archive',))
								return

						new_gallery.title = parsed['title']
						new_gallery.path = temp_p
						new_gallery.artist = parsed['artist']
						new_gallery.language = parsed['language']
						new_gallery.info = ""
						metafile.apply_gallery(new_gallery)
						if app_constants.MOVE_IMPORTED_GALLERIES and not app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
							new_gallery.path = utils.move_files(temp_p)

						self.data.append(new_gallery)
						log_i('Gallery successful created: {}'.format(folder_name.encode('utf-8', 'ignore')))
					else:
						log_i('Gallery already exists: {}'.format(folder_name.encode('utf-8', 'ignore')))
						self.skipped_paths.append((temp_p, 'Already exists'))
Exemple #13
0
    def local(self, s_path=None):
        """
        Do a local search in the given series_path.
        """
        self._data.clear()
        if s_path:
            self.series_path = s_path
        try:
            gallery_l = sorted([p.name for p in scandir.scandir(self.series_path)]) #list of folders in the "Gallery" folder
            mixed = False
        except TypeError:
            gallery_l = self.series_path
            mixed = True
        if len(gallery_l) != 0: # if gallery path list is not empty
            log_i('Gallery folder is not empty')
            if len(self.galleries_from_db) != len(app_constants.GALLERY_DATA):
                self._refresh_filter_list()
            self.DATA_COUNT.emit(len(gallery_l)) #tell model how many items are going to be added
            log_i('Received {} paths'.format(len(gallery_l)))
            progress = 0

            for folder_name in gallery_l: # folder_name = gallery folder title
                self._curr_gallery = folder_name
                if mixed:
                    path = folder_name
                    folder_name = os.path.split(path)[1]
                else:
                    path = os.path.join(self.series_path, folder_name)
                if app_constants.SUBFOLDER_AS_GALLERY or app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                    if app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                        app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY = False
                    log_i("Treating each subfolder as gallery")
                    if os.path.isdir(path):
                        gallery_folders, gallery_archives = utils.recursive_gallery_check(path)
                        for gs in gallery_folders:
                                self.create_gallery(gs, os.path.split(gs)[1], False)
                        p_saving = {}
                        for gs in gallery_archives:
                                    
                                self.create_gallery(gs[0], os.path.split(gs[0])[1], False, archive=gs[1])
                    elif path.endswith(utils.ARCHIVE_FILES):
                        for g in utils.check_archive(path):
                            self.create_gallery(g, os.path.split(g)[1], False, archive=path)
                else:
                    try:
                        if os.path.isdir(path):
                            if not list(scandir.scandir(path)):
                                raise ValueError
                        elif not path.endswith(utils.ARCHIVE_FILES):
                            raise NotADirectoryError

                        log_i("Treating each subfolder as chapter")
                        self.create_gallery(path, folder_name, do_chapters=True)

                    except ValueError:
                        self.skipped_paths.append((path, 'Empty directory'))
                        log_w('Directory is empty: {}'.format(path.encode(errors='ignore')))
                    except NotADirectoryError:
                        self.skipped_paths.append((path, 'Unsupported file'))
                        log_w('Unsupported file: {}'.format(path.encode(errors='ignore')))

                progress += 1 # update the progress bar
                self.PROGRESS.emit(progress)
        else: # if gallery folder is empty
            log_e('Local search error: Invalid directory')
            log_e('Gallery folder is empty')
            app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True # sanity check
            self.FINISHED.emit(False)
            # might want to include an error message
        app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = False
        # everything went well
        log_i('Local search: OK')
        log_i('Created {} items'.format(len(self._data)))
        if self._to_queue_container:
            for x in self._data:
                self._galleries_queue.put(x)
        else:
            self.FINISHED.emit(self._data)
        if self.skipped_paths:
            self.SKIPPED.emit(self.skipped_paths)
Exemple #14
0
    def local(self, s_path=None):
        """
		Do a local search in the given series_path.
		"""
        self.data.clear()
        if s_path:
            self.series_path = s_path
        try:
            gallery_l = sorted([
                p.name for p in scandir.scandir(self.series_path)
            ])  #list of folders in the "Gallery" folder
            mixed = False
        except TypeError:
            gallery_l = self.series_path
            mixed = True
        if len(gallery_l) != 0:  # if gallery path list is not empty
            log_i('Gallery folder is not empty')
            try:
                if len(self.galleries_from_db) != len(
                        app_constants.GALLERY_DATA):
                    self._refresh_filter_list()
                self.DATA_COUNT.emit(
                    len(gallery_l
                        ))  #tell model how many items are going to be added
                log_i('Received {} paths'.format(len(gallery_l)))
                progress = 0

                def create_gallery(path,
                                   folder_name,
                                   do_chapters=True,
                                   archive=None):
                    is_archive = True if archive else False
                    temp_p = archive if is_archive else path
                    folder_name = folder_name or path if folder_name or path else os.path.split(
                        archive)[1]
                    if utils.check_ignore_list(
                            temp_p) and not GalleryDB.check_exists(
                                temp_p, self.galleries_from_db, False):
                        log_i('Creating gallery: {}'.format(
                            folder_name.encode('utf-8', 'ignore')))
                        new_gallery = Gallery()
                        images_paths = []
                        metafile = utils.GMetafile()
                        try:
                            con = scandir.scandir(
                                temp_p)  #all of content in the gallery folder
                            log_i('Gallery source is a directory')
                            chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
                                if do_chapters else [] #subfolders
                            # if gallery has chapters divided into sub folders
                            numb_of_chapters = len(chapters)
                            if numb_of_chapters != 0:
                                log_i('Gallery has {} chapters'.format(
                                    numb_of_chapters))
                                for ch in chapters:
                                    chap = new_gallery.chapters.create_chapter(
                                    )
                                    chap.title = utils.title_parser(
                                        ch)['title']
                                    chap.path = os.path.join(path, ch)
                                    chap.pages = len(
                                        list(scandir.scandir(chap.path)))
                                    metafile.update(utils.GMetafile(chap.path))

                            else:  #else assume that all images are in gallery folder
                                chap = new_gallery.chapters.create_chapter()
                                chap.title = utils.title_parser(
                                    os.path.split(path)[1])['title']
                                chap.path = path
                                metafile.update(utils.GMetafile(chap.path))
                                chap.pages = len(list(scandir.scandir(path)))

                            parsed = utils.title_parser(folder_name)
                        except NotADirectoryError:
                            try:
                                if is_archive or temp_p.endswith(
                                        utils.ARCHIVE_FILES):
                                    log_i('Gallery source is an archive')
                                    contents = utils.check_archive(temp_p)
                                    if contents:
                                        new_gallery.is_archive = 1
                                        new_gallery.path_in_archive = '' if not is_archive else path
                                        if folder_name.endswith('/'):
                                            folder_name = folder_name[:-1]
                                            fn = os.path.split(folder_name)
                                            folder_name = fn[1] or fn[2]
                                        folder_name = folder_name.replace(
                                            '/', '')
                                        if folder_name.endswith(
                                                utils.ARCHIVE_FILES):
                                            n = folder_name
                                            for ext in utils.ARCHIVE_FILES:
                                                n = n.replace(ext, '')
                                            parsed = utils.title_parser(n)
                                        else:
                                            parsed = utils.title_parser(
                                                folder_name)

                                        if do_chapters:
                                            archive_g = sorted(contents)
                                            if not archive_g:
                                                log_w(
                                                    'No chapters found for {}'.
                                                    format(
                                                        temp_p.encode(
                                                            errors='ignore')))
                                                raise ValueError
                                            for g in archive_g:
                                                chap = new_gallery.chapters.create_chapter(
                                                )
                                                chap.in_archive = 1
                                                chap.title = utils.title_parser(
                                                    g)['title']
                                                chap.path = g
                                                metafile.update(
                                                    utils.GMetafile(g, temp_p))
                                                arch = utils.ArchiveFile(
                                                    temp_p)
                                                chap.pages = len(
                                                    arch.dir_contents(g))
                                                arch.close()
                                        else:
                                            chap = new_gallery.chapters.create_chapter(
                                            )
                                            chap.title = utils.title_parser(
                                                os.path.split(path)
                                                [1])['title']
                                            chap.in_archive = 1
                                            chap.path = path
                                            metafile.update(
                                                utils.GMetafile(path, temp_p))
                                            arch = utils.ArchiveFile(temp_p)
                                            chap.pages = len(
                                                arch.dir_contents(''))
                                            arch.close()
                                    else:
                                        raise ValueError
                                else:
                                    raise ValueError
                            except ValueError:
                                log_w('Skipped {} in local search'.format(
                                    path.encode(errors='ignore')))
                                self.skipped_paths.append((
                                    temp_p,
                                    'Empty archive',
                                ))
                                return
                            except app_constants.CreateArchiveFail:
                                log_w('Skipped {} in local search'.format(
                                    path.encode(errors='ignore')))
                                self.skipped_paths.append((
                                    temp_p,
                                    'Error creating archive',
                                ))
                                return

                        new_gallery.title = parsed['title']
                        new_gallery.path = temp_p
                        new_gallery.artist = parsed['artist']
                        new_gallery.language = parsed['language']
                        new_gallery.info = ""
                        metafile.apply_gallery(new_gallery)
                        if app_constants.MOVE_IMPORTED_GALLERIES and not app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
                            new_gallery.path = utils.move_files(temp_p)

                        self.data.append(new_gallery)
                        log_i('Gallery successful created: {}'.format(
                            folder_name.encode('utf-8', 'ignore')))
                    else:
                        log_i('Gallery already exists: {}'.format(
                            folder_name.encode('utf-8', 'ignore')))
                        self.skipped_paths.append((temp_p, 'Already exists'))

                for folder_name in gallery_l:  # folder_name = gallery folder title
                    self._curr_gallery = folder_name
                    if mixed:
                        path = folder_name
                        folder_name = os.path.split(path)[1]
                    else:
                        path = os.path.join(self.series_path, folder_name)
                    if app_constants.SUBFOLDER_AS_GALLERY or app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                        if app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
                            app_constants.OVERRIDE_SUBFOLDER_AS_GALLERY = False
                        log_i("Treating each subfolder as gallery")
                        if os.path.isdir(path):
                            gallery_folders, gallery_archives = utils.recursive_gallery_check(
                                path)
                            for gs in gallery_folders:
                                create_gallery(gs, os.path.split(gs)[1], False)
                            p_saving = {}
                            for gs in gallery_archives:

                                create_gallery(gs[0],
                                               os.path.split(gs[0])[1],
                                               False,
                                               archive=gs[1])
                        elif path.endswith(utils.ARCHIVE_FILES):
                            for g in utils.check_archive(path):
                                create_gallery(g,
                                               os.path.split(g)[1],
                                               False,
                                               archive=path)
                    else:
                        try:
                            if os.path.isdir(path):
                                if not list(scandir.scandir(path)):
                                    raise ValueError
                            elif not path.endswith(utils.ARCHIVE_FILES):
                                raise NotADirectoryError

                            log_i("Treating each subfolder as chapter")
                            create_gallery(path, folder_name, do_chapters=True)

                        except ValueError:
                            self.skipped_paths.append(
                                (path, 'Empty directory'))
                            log_w('Directory is empty: {}'.format(
                                path.encode(errors='ignore')))
                        except NotADirectoryError:
                            self.skipped_paths.append(
                                (path, 'Unsupported file'))
                            log_w('Unsupported file: {}'.format(
                                path.encode(errors='ignore')))

                    progress += 1  # update the progress bar
                    self.PROGRESS.emit(progress)
            except:
                log.exception('Local Search Error:')
                app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True  # sanity check
                self.FINISHED.emit(False)
        else:  # if gallery folder is empty
            log_e('Local search error: Invalid directory')
            log_e('Gallery folder is empty')
            app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True  # sanity check
            self.FINISHED.emit(False)
            # might want to include an error message
        app_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = False
        # everything went well
        log_i('Local search: OK')
        log_i('Created {} items'.format(len(self.data)))
        self.FINISHED.emit(self.data)
        if self.skipped_paths:
            self.SKIPPED.emit(self.skipped_paths)
Exemple #15
0
	def local(self, s_path=None):
		"""
		Do a local search in the given series_path.
		"""
		self.data.clear()
		if s_path:
			self.series_path = s_path
		try:
			gallery_l = sorted([p.name for p in scandir.scandir(self.series_path)]) #list of folders in the "Gallery" folder
			mixed = False
		except TypeError:
			gallery_l = self.series_path
			mixed = True
		if len(gallery_l) != 0: # if gallery path list is not empty
			log_i('Gallery folder is not empty')
			try:
				if len(self.galleries_from_db) != len(gui_constants.GALLERY_DATA):
					self._refresh_filter_list()
				self.DATA_COUNT.emit(len(gallery_l)) #tell model how many items are going to be added
				log_i('Found {} items'.format(len(gallery_l)))
				progress = 0
				def create_gallery(path, folder_name, do_chapters=True, archive=None):
					is_archive = True if archive else False
					temp_p = archive if is_archive else path
					folder_name = folder_name or path if folder_name or path else os.path.split(archive)[1]
					if utils.check_ignore_list(temp_p) and not GalleryDB.check_exists(temp_p, self.galleries_from_db, False):
						log_i('Creating gallery: {}'.format(folder_name.encode('utf-8', 'ignore')))
						new_gallery = Gallery()
						images_paths = []
						try:
							con = scandir.scandir(temp_p) #all of content in the gallery folder
							log_i('Gallery source is a directory')
							chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
							    if do_chapters else [] #subfolders
							# if gallery has chapters divided into sub folders
							if len(chapters) != 0:
								log_i('Gallery has chapters divided in directories')
								for numb, ch in enumerate(chapters):
									chap_path = os.path.join(path, ch)
									new_gallery.chapters[numb] = chap_path

							else: #else assume that all images are in gallery folder
								new_gallery.chapters[0] = path
				
							##find last edited file
							#times = set()
							#for root, dirs, files in os.walk(path, topdown=False):
							#	for img in files:
							#		fp = os.path.join(root, img)
							#		times.add( os.path.getmtime(fp) )
							#last_updated = time.asctime(time.gmtime(max(times)))
							#new_gallery.last_update = last_updated
							parsed = utils.title_parser(folder_name)
						except NotADirectoryError:
							try:
								if is_archive or temp_p.endswith(utils.ARCHIVE_FILES):
									log_i('Gallery source is an archive')
									contents = utils.check_archive(temp_p)
									if contents:
										new_gallery.is_archive = 1
										new_gallery.path_in_archive = '' if not is_archive else path
										if folder_name.endswith('/'):
											folder_name = folder_name[:-1]
											fn = os.path.split(folder_name)
											folder_name = fn[1] or fn[2]
										folder_name = folder_name.replace('/','')
										if folder_name.endswith(utils.ARCHIVE_FILES):
											n = folder_name
											for ext in utils.ARCHIVE_FILES:
												n = n.replace(ext, '')
											parsed = utils.title_parser(n)
										else:
											parsed = utils.title_parser(folder_name)
										if do_chapters:
											archive_g = sorted(contents)
											if not archive_g:
												log_w('No chapters found for {}'.format(temp_p.encode(errors='ignore')))
												raise ValueError
											for n, g in enumerate(archive_g):
												new_gallery.chapters[n] = g
										else:
											new_gallery.chapters[0] = path
									else:
										raise ValueError
								else:
									raise ValueError
							except ValueError:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append(temp_p)
								return

						new_gallery.title = parsed['title']
						new_gallery.path = temp_p
						new_gallery.artist = parsed['artist']
						new_gallery.language = parsed['language']
						new_gallery.info = "No description.."
						if gui_constants.MOVE_IMPORTED_GALLERIES and not gui_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
							new_gallery.path = utils.move_files(temp_p)

						self.data.append(new_gallery)
						log_i('Gallery successful created: {}'.format(folder_name.encode('utf-8', 'ignore')))
					else:
						log_i('Gallery already exists: {}'.format(folder_name.encode('utf-8', 'ignore')))
						self.skipped_paths.append(temp_p)

				for folder_name in gallery_l: # folder_name = gallery folder title
					self._curr_gallery = folder_name
					if mixed:
						path = folder_name
						folder_name = os.path.split(path)[1]
					else:
						path = os.path.join(self.series_path, folder_name)
					if gui_constants.SUBFOLDER_AS_GALLERY or gui_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
						if gui_constants.OVERRIDE_SUBFOLDER_AS_GALLERY:
							gui_constants.OVERRIDE_SUBFOLDER_AS_GALLERY = False
						log_i("Treating each subfolder as gallery")
						if os.path.isdir(path):
							gallery_folders, gallery_archives = utils.recursive_gallery_check(path)
							for gs in gallery_folders:
									create_gallery(gs, os.path.split(gs)[1], False)
							p_saving = {}
							for gs in gallery_archives:
									
									create_gallery(gs[0], os.path.split(gs[0])[1], False, archive=gs[1])
						elif path.endswith(utils.ARCHIVE_FILES):
							for g in utils.check_archive(path):
								create_gallery(g, os.path.split(g)[1], False, archive=path)
					else:
						if (os.path.isdir(path) and os.listdir(path)) or path.endswith(utils.ARCHIVE_FILES):
							log_i("Treating each subfolder as chapter")
							create_gallery(path, folder_name, do_chapters=True)
						else:
							self.skipped_paths.append(path)
							log_w('Directory is empty: {}'.format(path.encode(errors='ignore')))

					progress += 1 # update the progress bar
					self.PROGRESS.emit(progress)
			except:
				log.exception('Local Search Error:')
				gui_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True # sanity check
				self.FINISHED.emit(False)
		else: # if gallery folder is empty
			log_e('Local search error: Invalid directory')
			log_e('Gallery folder is empty')
			gui_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = True # sanity check
			self.FINISHED.emit(False)
			# might want to include an error message
		gui_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH = False
		# everything went well
		log_i('Local search: OK')
		log_i('Created {} items'.format(len(self.data)))
		self.FINISHED.emit(self.data)
		if self.skipped_paths:
			self.SKIPPED.emit(self.skipped_paths)
Exemple #16
0
				def create_gallery(path, folder_name, do_chapters=True, archive=None):
					is_archive = True if archive else False
					temp_p = archive if is_archive else path
					folder_name = folder_name or path if folder_name or path else os.path.split(archive)[1]
					if utils.check_ignore_list(temp_p) and not GalleryDB.check_exists(temp_p, self.galleries_from_db, False):
						log_i('Creating gallery: {}'.format(folder_name.encode('utf-8', 'ignore')))
						new_gallery = Gallery()
						images_paths = []
						try:
							con = scandir.scandir(temp_p) #all of content in the gallery folder
							log_i('Gallery source is a directory')
							chapters = sorted([sub.path for sub in con if sub.is_dir() or sub.name.endswith(utils.ARCHIVE_FILES)])\
							    if do_chapters else [] #subfolders
							# if gallery has chapters divided into sub folders
							if len(chapters) != 0:
								log_i('Gallery has chapters divided in directories')
								for numb, ch in enumerate(chapters):
									chap_path = os.path.join(path, ch)
									new_gallery.chapters[numb] = chap_path

							else: #else assume that all images are in gallery folder
								new_gallery.chapters[0] = path
				
							##find last edited file
							#times = set()
							#for root, dirs, files in os.walk(path, topdown=False):
							#	for img in files:
							#		fp = os.path.join(root, img)
							#		times.add( os.path.getmtime(fp) )
							#last_updated = time.asctime(time.gmtime(max(times)))
							#new_gallery.last_update = last_updated
							parsed = utils.title_parser(folder_name)
						except NotADirectoryError:
							try:
								if is_archive or temp_p.endswith(utils.ARCHIVE_FILES):
									log_i('Gallery source is an archive')
									contents = utils.check_archive(temp_p)
									if contents:
										new_gallery.is_archive = 1
										new_gallery.path_in_archive = '' if not is_archive else path
										if folder_name.endswith('/'):
											folder_name = folder_name[:-1]
											fn = os.path.split(folder_name)
											folder_name = fn[1] or fn[2]
										folder_name = folder_name.replace('/','')
										if folder_name.endswith(utils.ARCHIVE_FILES):
											n = folder_name
											for ext in utils.ARCHIVE_FILES:
												n = n.replace(ext, '')
											parsed = utils.title_parser(n)
										else:
											parsed = utils.title_parser(folder_name)
										if do_chapters:
											archive_g = sorted(contents)
											if not archive_g:
												log_w('No chapters found for {}'.format(temp_p.encode(errors='ignore')))
												raise ValueError
											for n, g in enumerate(archive_g):
												new_gallery.chapters[n] = g
										else:
											new_gallery.chapters[0] = path
									else:
										raise ValueError
								else:
									raise ValueError
							except ValueError:
								log_w('Skipped {} in local search'.format(path.encode(errors='ignore')))
								self.skipped_paths.append(temp_p)
								return

						new_gallery.title = parsed['title']
						new_gallery.path = temp_p
						new_gallery.artist = parsed['artist']
						new_gallery.language = parsed['language']
						new_gallery.info = "No description.."
						if gui_constants.MOVE_IMPORTED_GALLERIES and not gui_constants.OVERRIDE_MOVE_IMPORTED_IN_FETCH:
							new_gallery.path = utils.move_files(temp_p)

						self.data.append(new_gallery)
						log_i('Gallery successful created: {}'.format(folder_name.encode('utf-8', 'ignore')))
					else:
						log_i('Gallery already exists: {}'.format(folder_name.encode('utf-8', 'ignore')))
						self.skipped_paths.append(temp_p)