Total Complexity | 45 |
Total Lines | 372 |
Duplicated Lines | 0 % |
Complex classes like LinterBase often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
1 | from contextlib import contextmanager |
||
302 | class LinterBase(LocalBear, metaclass=LinterMeta): |
||
|
|||
303 | |||
304 | @staticmethod |
||
305 | def generate_config(filename, file): |
||
306 | """ |
||
307 | Generates the content of a config-file the linter-tool might |
||
308 | need. |
||
309 | |||
310 | The contents generated from this function are written to a |
||
311 | temporary file and the path is provided inside |
||
312 | ``create_arguments()``. |
||
313 | |||
314 | By default no configuration is generated. |
||
315 | |||
316 | You can provide additional keyword arguments and defaults. |
||
317 | These will be interpreted as required settings that need to be |
||
318 | provided through a coafile-section. |
||
319 | |||
320 | :param filename: |
||
321 | The name of the file currently processed. |
||
322 | :param file: |
||
323 | The contents of the file currently processed. |
||
324 | :return: |
||
325 | The config-file-contents as a string or ``None``. |
||
326 | """ |
||
327 | return None |
||
328 | |||
329 | @staticmethod |
||
330 | def create_arguments(filename, file, config_file): |
||
331 | """ |
||
332 | Creates the arguments for the linter. |
||
333 | |||
334 | You can provide additional keyword arguments and defaults. |
||
335 | These will be interpreted as required settings that need to be |
||
336 | provided through a coafile-section. |
||
337 | |||
338 | :param filename: |
||
339 | The name of the file the linter-tool shall process. |
||
340 | :param file: |
||
341 | The contents of the file. |
||
342 | :param config_file: |
||
343 | The path of the config-file if used. ``None`` if unused. |
||
344 | :return: |
||
345 | A sequence of arguments to feed the linter-tool with. |
||
346 | """ |
||
347 | raise NotImplementedError |
||
348 | |||
349 | @staticmethod |
||
350 | def get_executable(): |
||
351 | """ |
||
352 | Returns the executable of this class. |
||
353 | |||
354 | :return: |
||
355 | The executable name. |
||
356 | """ |
||
357 | return options["executable"] |
||
358 | |||
359 | @classmethod |
||
360 | def check_prerequisites(cls): |
||
361 | """ |
||
362 | Checks whether the linter-tool the bear uses is operational. |
||
363 | |||
364 | :return: |
||
365 | True if available, otherwise a string containing more info. |
||
366 | """ |
||
367 | if shutil.which(cls.get_executable()) is None: |
||
368 | return repr(cls.get_executable()) + " is not installed." |
||
369 | else: |
||
370 | if options["prerequisite_check_command"]: |
||
371 | try: |
||
372 | check_call(options["prerequisite_check_command"], |
||
373 | stdout=DEVNULL, |
||
374 | stderr=DEVNULL) |
||
375 | return True |
||
376 | except (OSError, CalledProcessError): |
||
377 | return options["prerequisite_check_fail_message"] |
||
378 | return True |
||
379 | |||
380 | @classmethod |
||
381 | def _get_create_arguments_metadata(cls): |
||
382 | return FunctionMetadata.from_function( |
||
383 | cls.create_arguments, |
||
384 | omit={"filename", "file", "config_file"}) |
||
385 | |||
386 | @classmethod |
||
387 | def _get_generate_config_metadata(cls): |
||
388 | return FunctionMetadata.from_function( |
||
389 | cls.generate_config, |
||
390 | omit={"filename", "file"}) |
||
391 | |||
392 | @classmethod |
||
393 | def _get_process_output_metadata(cls): |
||
394 | return FunctionMetadata.from_function( |
||
395 | cls.process_output, |
||
396 | omit={"self", "output", "filename", "file"}) |
||
397 | |||
398 | @classmethod |
||
399 | def get_non_optional_settings(cls): |
||
400 | return cls.get_metadata().non_optional_params |
||
401 | |||
402 | @classmethod |
||
403 | def get_metadata(cls): |
||
404 | merged_metadata = FunctionMetadata.merge( |
||
405 | cls._get_process_output_metadata(), |
||
406 | cls._get_generate_config_metadata(), |
||
407 | cls._get_create_arguments_metadata()) |
||
408 | merged_metadata.desc = inspect.getdoc(cls) |
||
409 | return merged_metadata |
||
410 | |||
411 | @classmethod |
||
412 | def _execute_command(cls, args, stdin=None): |
||
413 | """ |
||
414 | Executes the underlying tool with the given arguments. |
||
415 | |||
416 | :param args: |
||
417 | The argument sequence to pass to the executable. |
||
418 | :param stdin: |
||
419 | Input to send to the opened process as stdin. |
||
420 | :return: |
||
421 | A tuple with ``(stdout, stderr)``. |
||
422 | """ |
||
423 | return run_shell_command( |
||
424 | (cls.get_executable(),) + tuple(args), |
||
425 | stdin=stdin) |
||
426 | |||
427 | def _convert_output_regex_match_to_result(self, |
||
428 | match, |
||
429 | filename, |
||
430 | severity_map): |
||
431 | """ |
||
432 | Converts the matched named-groups of ``output_regex`` to an |
||
433 | actual ``Result``. |
||
434 | |||
435 | :param match: |
||
436 | The regex match object. |
||
437 | :param filename: |
||
438 | The name of the file this match belongs to. |
||
439 | :param severity_map: |
||
440 | The dict to use to map the severity-match to an actual |
||
441 | ``RESULT_SEVERITY``. |
||
442 | """ |
||
443 | # Pre process the groups |
||
444 | groups = match.groupdict() |
||
445 | |||
446 | try: |
||
447 | groups["severity"] = severity_map[ |
||
448 | groups["severity"].lower()] |
||
449 | except KeyError: |
||
450 | self.warn( |
||
451 | "No correspondence for " + repr(groups["severity"]) + |
||
452 | " found in given severity map. Assuming " |
||
453 | "`RESULT_SEVERITY.NORMAL`.") |
||
454 | groups["severity"] = RESULT_SEVERITY.NORMAL |
||
455 | |||
456 | for variable in ("line", "column", "end_line", "end_column"): |
||
457 | groups[variable] = (None |
||
458 | if groups.get(variable, "") == "" else |
||
459 | int(groups[variable])) |
||
460 | |||
461 | if "origin" in groups: |
||
462 | groups["origin"] = "{} ({})".format( |
||
463 | str(klass.__name__), |
||
464 | str(groups["origin"])) |
||
465 | |||
466 | # Construct the result. |
||
467 | return Result.from_values( |
||
468 | origin=groups.get("origin", self), |
||
469 | message=groups.get("message", ""), |
||
470 | file=filename, |
||
471 | severity=int(groups.get("severity", |
||
472 | RESULT_SEVERITY.NORMAL)), |
||
473 | line=groups["line"], |
||
474 | column=groups["column"], |
||
475 | end_line=groups["end_line"], |
||
476 | end_column=groups["end_column"]) |
||
477 | |||
478 | def process_output_corrected(self, |
||
479 | output, |
||
480 | filename, |
||
481 | file, |
||
482 | diff_severity=RESULT_SEVERITY.NORMAL, |
||
483 | diff_message="Inconsistency found."): |
||
484 | """ |
||
485 | Processes the executable's output as a corrected file. |
||
486 | |||
487 | :param output: |
||
488 | The output of the program. This can be either a single |
||
489 | string or a sequence of strings. |
||
490 | :param filename: |
||
491 | The filename of the file currently being corrected. |
||
492 | :param file: |
||
493 | The contents of the file currently being corrected. |
||
494 | :param diff_severity: |
||
495 | The severity to use for generating results. |
||
496 | :param diff_message: |
||
497 | The message to use for generating results. |
||
498 | :return: |
||
499 | An iterator returning results containing patches for the |
||
500 | file to correct. |
||
501 | """ |
||
502 | if isinstance(output, str): |
||
503 | output = (output,) |
||
504 | |||
505 | for string in output: |
||
506 | for diff in Diff.from_string_arrays( |
||
507 | file, |
||
508 | string.splitlines(keepends=True)).split_diff(): |
||
509 | yield Result(self, |
||
510 | diff_message, |
||
511 | affected_code=(diff.range(filename),), |
||
512 | diffs={filename: diff}, |
||
513 | severity=diff_severity) |
||
514 | |||
515 | def process_output_regex( |
||
516 | self, |
||
517 | output, |
||
518 | filename, |
||
519 | file, |
||
520 | output_regex, |
||
521 | severity_map=MappingProxyType({ |
||
522 | "error": RESULT_SEVERITY.MAJOR, |
||
523 | "warning": RESULT_SEVERITY.NORMAL, |
||
524 | "warn": RESULT_SEVERITY.NORMAL, |
||
525 | "info": RESULT_SEVERITY.INFO})): |
||
526 | """ |
||
527 | Processes the executable's output using a regex. |
||
528 | |||
529 | :param output: |
||
530 | The output of the program. This can be either a single |
||
531 | string or a sequence of strings. |
||
532 | :param filename: |
||
533 | The filename of the file currently being corrected. |
||
534 | :param file: |
||
535 | The contents of the file currently being corrected. |
||
536 | :param output_regex: |
||
537 | The regex to parse the output with. It should use as many |
||
538 | of the following named groups (via ``(?P<name>...)``) to |
||
539 | provide a good result: |
||
540 | |||
541 | - line - The line where the issue starts. |
||
542 | - column - The column where the issue starts. |
||
543 | - end_line - The line where the issue ends. |
||
544 | - end_column - The column where the issue ends. |
||
545 | - severity - The severity of the issue. |
||
546 | - message - The message of the result. |
||
547 | - origin - The origin of the issue. |
||
548 | |||
549 | The groups ``line``, ``column``, ``end_line`` and |
||
550 | ``end_column`` don't have to match numbers only, they can |
||
551 | also match nothing, the generated ``Result`` is filled |
||
552 | automatically with ``None`` then for the appropriate |
||
553 | properties. |
||
554 | :param severity_map: |
||
555 | A dict used to map a severity string (captured from the |
||
556 | ``output_regex`` with the named group ``severity``) to an |
||
557 | actual ``coalib.results.RESULT_SEVERITY`` for a result. |
||
558 | :return: |
||
559 | An iterator returning results. |
||
560 | """ |
||
561 | if isinstance(output, str): |
||
562 | output = (output,) |
||
563 | |||
564 | for string in output: |
||
565 | for match in re.finditer(output_regex, string): |
||
566 | yield self._convert_output_regex_match_to_result( |
||
567 | match, filename, severity_map=severity_map) |
||
568 | |||
569 | if options["output_format"] is None: |
||
570 | # Check if user supplied a `process_output` override. |
||
571 | if not (hasattr(klass, "process_output") and |
||
572 | callable(klass.process_output)): |
||
573 | raise ValueError("`process_output` not provided by given " |
||
574 | "class.") |
||
575 | # No need to assign to `process_output` here, the class mixing |
||
576 | # below automatically does that. |
||
577 | else: |
||
578 | # Prevent people from accidentally defining `process_output` |
||
579 | # manually, as this would implicitly override the internally |
||
580 | # set-up `process_output`. |
||
581 | if hasattr(klass, "process_output"): |
||
582 | raise ValueError("`process_output` is used by given class," |
||
583 | " but " + repr(options["output_format"]) + |
||
584 | " output format was specified.") |
||
585 | |||
586 | if options["output_format"] == "corrected": |
||
587 | process_output_args = {} |
||
588 | if "diff_severity" in options: |
||
589 | process_output_args["diff_severity"] = ( |
||
590 | options["diff_severity"]) |
||
591 | if "diff_message" in options: |
||
592 | process_output_args["diff_message"] = ( |
||
593 | options["diff_message"]) |
||
594 | |||
595 | process_output = partialmethod( |
||
596 | process_output_corrected, **process_output_args) |
||
597 | |||
598 | elif options["output_format"] == "regex": |
||
599 | process_output_args = { |
||
600 | "output_regex": options["output_regex"]} |
||
601 | if "severity_map" in options: |
||
602 | process_output_args["severity_map"] = ( |
||
603 | options["severity_map"]) |
||
604 | |||
605 | process_output = partialmethod( |
||
606 | process_output_regex, **process_output_args) |
||
607 | |||
608 | else: # pragma: no cover |
||
609 | # This statement is never reached. |
||
610 | assert False |
||
611 | |||
612 | @classmethod |
||
613 | @contextmanager |
||
614 | def _create_config(cls, filename, file, **kwargs): |
||
615 | """ |
||
616 | Provides a context-manager that creates the config file if the |
||
617 | user provides one and cleans it up when done with linting. |
||
618 | |||
619 | :param filename: |
||
620 | The filename of the file. |
||
621 | :param file: |
||
622 | The file contents. |
||
623 | :param kwargs: |
||
624 | Section settings passed from ``run()``. |
||
625 | :return: |
||
626 | A context-manager handling the config-file. |
||
627 | """ |
||
628 | content = cls.generate_config(filename, file, **kwargs) |
||
629 | if content is None: |
||
630 | yield None |
||
631 | else: |
||
632 | tmp_suffix = options["config_suffix"] |
||
633 | with make_temp(suffix=tmp_suffix) as config_file: |
||
634 | with open(config_file, mode="w") as fl: |
||
635 | fl.write(content) |
||
636 | yield config_file |
||
637 | |||
638 | def run(self, filename, file, **kwargs): |
||
639 | # Get the **kwargs params to forward to `generate_config()` |
||
640 | # (from `_create_config()`). |
||
641 | generate_config_kwargs = FunctionMetadata.filter_parameters( |
||
642 | self._get_generate_config_metadata(), kwargs) |
||
643 | |||
644 | with self._create_config( |
||
645 | filename, |
||
646 | file, |
||
647 | **generate_config_kwargs) as config_file: |
||
648 | |||
649 | # And now retrieve the **kwargs for `create_arguments()`. |
||
650 | create_arguments_kwargs = ( |
||
651 | FunctionMetadata.filter_parameters( |
||
652 | self._get_create_arguments_metadata(), kwargs)) |
||
653 | |||
654 | output = self._execute_command( |
||
655 | self.create_arguments(filename, |
||
656 | file, |
||
657 | config_file, |
||
658 | **create_arguments_kwargs), |
||
659 | stdin="".join(file) if options["use_stdin"] else None) |
||
660 | output = tuple(compress( |
||
661 | output, |
||
662 | (options["use_stdout"], options["use_stderr"]))) |
||
663 | if len(output) == 1: |
||
664 | output = output[0] |
||
665 | |||
666 | process_output_kwargs = FunctionMetadata.filter_parameters( |
||
667 | self._get_process_output_metadata(), kwargs) |
||
668 | return self.process_output(output, filename, file, |
||
669 | **process_output_kwargs) |
||
670 | |||
671 | def __repr__(self): |
||
672 | return "<{} linter object (wrapping {})>".format( |
||
673 | type(self).__name__, repr(options["executable"])) |
||
674 | |||
681 |