saucenao.worker

Complexity

Total Complexity

21

Size/Duplication

Total Lines	129
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
B	Worker.move_to_categories()	0	33	5
B	Worker.run()	0	26	5
A	Worker.excludes()	0	10	3
A	Worker.files()	0	14	3
A	Worker.get_similar_title()	0	16	4
A	Worker.__init__()	0	10	1

#!/usr/bin/python

This module should have a docstring.

# -*- coding: utf-8 -*-
import time

from titlesearch import get_similar_titles

Unable to import 'titlesearch'

from saucenao import SauceNao, FileHandler


class Worker(SauceNao):

The variable __class__ seems to be unused.

    """

Found indentation with spaces instead of tabs

    Worker class for checking a list of files
    """

    def __init__(self, files, *args, **kwargs):

Found indentation with spaces instead of tabs

        """

Found indentation with spaces instead of tabs

        initializing function

        :type files: list|tuple|Generator
        :param args:
        :param kwargs:
        """
        super().__init__(*args, **kwargs)

Found indentation with spaces instead of tabs

        self.complete_file_list = files

Found indentation with spaces instead of tabs

    def run(self):

Found indentation with spaces instead of tabs

        """Check all files with SauceNao and execute the specified tasks

Found indentation with spaces instead of tabs

        :return:
        """
        for file_name in self.files:

Found indentation with spaces instead of tabs

            start_time = time.time()

Found indentation with spaces instead of tabs

            filtered_results = self.check_file(file_name)

Found indentation with spaces instead of tabs

            if not filtered_results:

Found indentation with spaces instead of tabs

                self.logger.info('No results found for image: {0:s}'.format(file_name))

This line is too long as per the coding-style (87/80).

Found indentation with spaces instead of tabs

                continue

Found indentation with spaces instead of tabs

            if self._move_to_categories:

Found indentation with spaces instead of tabs

                self.move_to_categories(file_name=file_name, results=filtered_results)

This line is too long as per the coding-style (86/80).

Found indentation with spaces instead of tabs

            else:

Found indentation with spaces instead of tabs

                yield {

Found indentation with spaces instead of tabs

                    'filename': file_name,
                    'results': filtered_results
                }

            duration = time.time() - start_time

Found indentation with spaces instead of tabs

            if duration < (30 / SauceNao.LIMIT_30_SECONDS):

Found indentation with spaces instead of tabs

                self.logger.debug("sleeping '{:.2f}' seconds".format((30 / SauceNao.LIMIT_30_SECONDS) - duration))

This line is too long as per the coding-style (114/80).

Found indentation with spaces instead of tabs

                time.sleep((30 / SauceNao.LIMIT_30_SECONDS) - duration)

Found indentation with spaces instead of tabs

    @property

Found indentation with spaces instead of tabs

    def excludes(self):

Found indentation with spaces instead of tabs

        """Property for excludes

Found indentation with spaces instead of tabs

        :return:
        """
        if self._exclude_categories:

Found indentation with spaces instead of tabs

Unnecessary "else" after "return"

            return [l.lower() for l in self._exclude_categories.split(",")]

Found indentation with spaces instead of tabs

        else:

Found indentation with spaces instead of tabs

            return []

Found indentation with spaces instead of tabs

    @property

Found indentation with spaces instead of tabs

    def files(self):

Found indentation with spaces instead of tabs

        """Property for files

Found indentation with spaces instead of tabs

        :return:
        """
        if self._start_file:

Found indentation with spaces instead of tabs

            # change files from generator to list
            files = list(self.complete_file_list)

Found indentation with spaces instead of tabs

            try:

Found indentation with spaces instead of tabs

                return files[files.index(self._start_file):]

Found indentation with spaces instead of tabs

            except ValueError:

Found indentation with spaces instead of tabs

                return self.complete_file_list

Found indentation with spaces instead of tabs

        return self.complete_file_list

Found indentation with spaces instead of tabs

    def move_to_categories(self, file_name: str, results):

Found indentation with spaces instead of tabs

        """Check the file for categories and move it to the corresponding folder

Found indentation with spaces instead of tabs

        :type file_name: str
        :type results: list|tuple|Generator
        :return: bool
        """
        categories = self.get_content_value(results, SauceNao.CONTENT_CATEGORY_KEY)

This line is too long as per the coding-style (83/80).

Found indentation with spaces instead of tabs

        if not categories:

Found indentation with spaces instead of tabs

            self.logger.info("no categories found for file: {0:s}".format(file_name))

This line is too long as per the coding-style (85/80).

Found indentation with spaces instead of tabs

            return False

Found indentation with spaces instead of tabs

        self.logger.debug('categories: {0:s}'.format(', '.join(categories)))

Found indentation with spaces instead of tabs

        # since many pictures are tagged as original and with a proper category
        # we remove the original category if we have more than 1 category
        if len(categories) > 1 and 'original' in categories:

Found indentation with spaces instead of tabs

            categories.remove('original')

Found indentation with spaces instead of tabs

        # take the first category
        category = categories[0]

Found indentation with spaces instead of tabs

        category = self.get_similar_title(category)

Found indentation with spaces instead of tabs

        # sub categories we don't want to move like original etc
        if category.lower() in self.excludes:

Found indentation with spaces instead of tabs

            self.logger.info("skipping excluded category: {0:s} ({1:s})".format(category, file_name))

This line is too long as per the coding-style (101/80).

Found indentation with spaces instead of tabs

            return False

Found indentation with spaces instead of tabs

        self.logger.info("moving {0:s} to category: {1:s}".format(file_name, category))

This line is too long as per the coding-style (87/80).

Found indentation with spaces instead of tabs

        FileHandler.move_to_category(file_name, category, base_directory=self._directory)

This line is too long as per the coding-style (89/80).

Found indentation with spaces instead of tabs

        return True

Found indentation with spaces instead of tabs

    def get_similar_title(self, category):

Found indentation with spaces instead of tabs

        """

Found indentation with spaces instead of tabs

        :param category:
        :return:
        """
        if get_similar_titles:

Found indentation with spaces instead of tabs

            similar_titles = get_similar_titles(category)

Found indentation with spaces instead of tabs

            if similar_titles and similar_titles[0]['similarity'] * 100 >= self._title_minimum_similarity:

This line is too long as per the coding-style (106/80).

Found indentation with spaces instead of tabs

                self.logger.info(

Found indentation with spaces instead of tabs

                    "Similar title found: {0:s}, {1:s} ({2:.2f}%)".format(
                        category, similar_titles[0]['title'], similar_titles[0]['similarity'] * 100))

This line is too long as per the coding-style (101/80).

                return similar_titles[0]['title']

Found indentation with spaces instead of tabs

        return category

Found indentation with spaces instead of tabs