misc: Add pylint
This commit is contained in:
parent
e31b0ebf8a
commit
1f3e5b28fe
596
.pylintrc
Normal file
596
.pylintrc
Normal file
@ -0,0 +1,596 @@
|
|||||||
|
[MASTER]
|
||||||
|
|
||||||
|
# A comma-separated list of package or module names from where C extensions may
|
||||||
|
# be loaded. Extensions are loading into the active Python interpreter and may
|
||||||
|
# run arbitrary code.
|
||||||
|
extension-pkg-whitelist=
|
||||||
|
|
||||||
|
# Specify a score threshold to be exceeded before program exits with error.
|
||||||
|
fail-under=10
|
||||||
|
|
||||||
|
# Add files or directories to the blacklist. They should be base names, not
|
||||||
|
# paths.
|
||||||
|
ignore=CVS
|
||||||
|
|
||||||
|
# Add files or directories matching the regex patterns to the blacklist. The
|
||||||
|
# regex matches against base names, not paths.
|
||||||
|
ignore-patterns=
|
||||||
|
|
||||||
|
# Python code to execute, usually for sys.path manipulation such as
|
||||||
|
# pygtk.require().
|
||||||
|
#init-hook=
|
||||||
|
|
||||||
|
# Use multiple processes to speed up Pylint. Specifying 0 will auto-detect the
|
||||||
|
# number of processors available to use.
|
||||||
|
jobs=1
|
||||||
|
|
||||||
|
# Control the amount of potential inferred values when inferring a single
|
||||||
|
# object. This can help the performance when dealing with large functions or
|
||||||
|
# complex, nested conditions.
|
||||||
|
limit-inference-results=100
|
||||||
|
|
||||||
|
# List of plugins (as comma separated values of python module names) to load,
|
||||||
|
# usually to register additional checkers.
|
||||||
|
load-plugins=
|
||||||
|
|
||||||
|
# Pickle collected data for later comparisons.
|
||||||
|
persistent=yes
|
||||||
|
|
||||||
|
# When enabled, pylint would attempt to guess common misconfiguration and emit
|
||||||
|
# user-friendly hints instead of false-positive error messages.
|
||||||
|
suggestion-mode=yes
|
||||||
|
|
||||||
|
# Allow loading of arbitrary C extensions. Extensions are imported into the
|
||||||
|
# active Python interpreter and may run arbitrary code.
|
||||||
|
unsafe-load-any-extension=no
|
||||||
|
|
||||||
|
|
||||||
|
[MESSAGES CONTROL]
|
||||||
|
|
||||||
|
# Only show warnings with the listed confidence levels. Leave empty to show
|
||||||
|
# all. Valid levels: HIGH, INFERENCE, INFERENCE_FAILURE, UNDEFINED.
|
||||||
|
confidence=
|
||||||
|
|
||||||
|
# Disable the message, report, category or checker with the given id(s). You
|
||||||
|
# can either give multiple identifiers separated by comma (,) or put this
|
||||||
|
# option multiple times (only on the command line, not in the configuration
|
||||||
|
# file where it should appear only once). You can also use "--disable=all" to
|
||||||
|
# disable everything first and then reenable specific checks. For example, if
|
||||||
|
# you want to run only the similarities checker, you can use "--disable=all
|
||||||
|
# --enable=similarities". If you want to run only the classes checker, but have
|
||||||
|
# no Warning level messages displayed, use "--disable=all --enable=classes
|
||||||
|
# --disable=W".
|
||||||
|
disable=print-statement,
|
||||||
|
parameter-unpacking,
|
||||||
|
unpacking-in-except,
|
||||||
|
old-raise-syntax,
|
||||||
|
backtick,
|
||||||
|
long-suffix,
|
||||||
|
old-ne-operator,
|
||||||
|
old-octal-literal,
|
||||||
|
import-star-module-level,
|
||||||
|
non-ascii-bytes-literal,
|
||||||
|
raw-checker-failed,
|
||||||
|
bad-inline-option,
|
||||||
|
locally-disabled,
|
||||||
|
file-ignored,
|
||||||
|
suppressed-message,
|
||||||
|
useless-suppression,
|
||||||
|
deprecated-pragma,
|
||||||
|
use-symbolic-message-instead,
|
||||||
|
apply-builtin,
|
||||||
|
basestring-builtin,
|
||||||
|
buffer-builtin,
|
||||||
|
cmp-builtin,
|
||||||
|
coerce-builtin,
|
||||||
|
execfile-builtin,
|
||||||
|
file-builtin,
|
||||||
|
long-builtin,
|
||||||
|
raw_input-builtin,
|
||||||
|
reduce-builtin,
|
||||||
|
standarderror-builtin,
|
||||||
|
unicode-builtin,
|
||||||
|
xrange-builtin,
|
||||||
|
coerce-method,
|
||||||
|
delslice-method,
|
||||||
|
getslice-method,
|
||||||
|
setslice-method,
|
||||||
|
no-absolute-import,
|
||||||
|
old-division,
|
||||||
|
dict-iter-method,
|
||||||
|
dict-view-method,
|
||||||
|
next-method-called,
|
||||||
|
metaclass-assignment,
|
||||||
|
indexing-exception,
|
||||||
|
raising-string,
|
||||||
|
reload-builtin,
|
||||||
|
oct-method,
|
||||||
|
hex-method,
|
||||||
|
nonzero-method,
|
||||||
|
cmp-method,
|
||||||
|
input-builtin,
|
||||||
|
round-builtin,
|
||||||
|
intern-builtin,
|
||||||
|
unichr-builtin,
|
||||||
|
map-builtin-not-iterating,
|
||||||
|
zip-builtin-not-iterating,
|
||||||
|
range-builtin-not-iterating,
|
||||||
|
filter-builtin-not-iterating,
|
||||||
|
using-cmp-argument,
|
||||||
|
eq-without-hash,
|
||||||
|
div-method,
|
||||||
|
idiv-method,
|
||||||
|
rdiv-method,
|
||||||
|
exception-message-attribute,
|
||||||
|
invalid-str-codec,
|
||||||
|
sys-max-int,
|
||||||
|
bad-python3-import,
|
||||||
|
deprecated-string-function,
|
||||||
|
deprecated-str-translate-call,
|
||||||
|
deprecated-itertools-function,
|
||||||
|
deprecated-types-field,
|
||||||
|
next-method-defined,
|
||||||
|
dict-items-not-iterating,
|
||||||
|
dict-keys-not-iterating,
|
||||||
|
dict-values-not-iterating,
|
||||||
|
deprecated-operator-function,
|
||||||
|
deprecated-urllib-function,
|
||||||
|
xreadlines-attribute,
|
||||||
|
deprecated-sys-function,
|
||||||
|
exception-escape,
|
||||||
|
comprehension-escape,
|
||||||
|
no-value-for-parameter
|
||||||
|
|
||||||
|
# Enable the message, report, category or checker with the given id(s). You can
|
||||||
|
# either give multiple identifier separated by comma (,) or put this option
|
||||||
|
# multiple time (only on the command line, not in the configuration file where
|
||||||
|
# it should appear only once). See also the "--disable" option for examples.
|
||||||
|
enable=c-extension-no-member
|
||||||
|
|
||||||
|
|
||||||
|
[REPORTS]
|
||||||
|
|
||||||
|
# Python expression which should return a score less than or equal to 10. You
|
||||||
|
# have access to the variables 'error', 'warning', 'refactor', and 'convention'
|
||||||
|
# which contain the number of messages in each category, as well as 'statement'
|
||||||
|
# which is the total number of statements analyzed. This score is used by the
|
||||||
|
# global evaluation report (RP0004).
|
||||||
|
evaluation=10.0 - ((float(5 * error + warning + refactor + convention) / statement) * 10)
|
||||||
|
|
||||||
|
# Template used to display messages. This is a python new-style format string
|
||||||
|
# used to format the message information. See doc for all details.
|
||||||
|
#msg-template=
|
||||||
|
|
||||||
|
# Set the output format. Available formats are text, parseable, colorized, json
|
||||||
|
# and msvs (visual studio). You can also give a reporter class, e.g.
|
||||||
|
# mypackage.mymodule.MyReporterClass.
|
||||||
|
output-format=text
|
||||||
|
|
||||||
|
# Tells whether to display a full report or only the messages.
|
||||||
|
reports=no
|
||||||
|
|
||||||
|
# Activate the evaluation score.
|
||||||
|
score=yes
|
||||||
|
|
||||||
|
|
||||||
|
[REFACTORING]
|
||||||
|
|
||||||
|
# Maximum number of nested blocks for function / method body
|
||||||
|
max-nested-blocks=5
|
||||||
|
|
||||||
|
# Complete name of functions that never returns. When checking for
|
||||||
|
# inconsistent-return-statements if a never returning function is called then
|
||||||
|
# it will be considered as an explicit return statement and no message will be
|
||||||
|
# printed.
|
||||||
|
never-returning-functions=sys.exit
|
||||||
|
|
||||||
|
|
||||||
|
[BASIC]
|
||||||
|
|
||||||
|
# Naming style matching correct argument names.
|
||||||
|
argument-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct argument names. Overrides argument-
|
||||||
|
# naming-style.
|
||||||
|
#argument-rgx=
|
||||||
|
|
||||||
|
# Naming style matching correct attribute names.
|
||||||
|
attr-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct attribute names. Overrides attr-naming-
|
||||||
|
# style.
|
||||||
|
#attr-rgx=
|
||||||
|
|
||||||
|
# Bad variable names which should always be refused, separated by a comma.
|
||||||
|
bad-names=foo,
|
||||||
|
bar,
|
||||||
|
baz,
|
||||||
|
toto,
|
||||||
|
tutu,
|
||||||
|
tata
|
||||||
|
|
||||||
|
# Bad variable names regexes, separated by a comma. If names match any regex,
|
||||||
|
# they will always be refused
|
||||||
|
bad-names-rgxs=
|
||||||
|
|
||||||
|
# Naming style matching correct class attribute names.
|
||||||
|
class-attribute-naming-style=any
|
||||||
|
|
||||||
|
# Regular expression matching correct class attribute names. Overrides class-
|
||||||
|
# attribute-naming-style.
|
||||||
|
#class-attribute-rgx=
|
||||||
|
|
||||||
|
# Naming style matching correct class names.
|
||||||
|
class-naming-style=PascalCase
|
||||||
|
|
||||||
|
# Regular expression matching correct class names. Overrides class-naming-
|
||||||
|
# style.
|
||||||
|
#class-rgx=
|
||||||
|
|
||||||
|
# Naming style matching correct constant names.
|
||||||
|
const-naming-style=UPPER_CASE
|
||||||
|
|
||||||
|
# Regular expression matching correct constant names. Overrides const-naming-
|
||||||
|
# style.
|
||||||
|
#const-rgx=
|
||||||
|
|
||||||
|
# Minimum line length for functions/classes that require docstrings, shorter
|
||||||
|
# ones are exempt.
|
||||||
|
docstring-min-length=-1
|
||||||
|
|
||||||
|
# Naming style matching correct function names.
|
||||||
|
function-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct function names. Overrides function-
|
||||||
|
# naming-style.
|
||||||
|
#function-rgx=
|
||||||
|
|
||||||
|
# Good variable names which should always be accepted, separated by a comma.
|
||||||
|
good-names=i,
|
||||||
|
j,
|
||||||
|
k,
|
||||||
|
ex,
|
||||||
|
Run,
|
||||||
|
_
|
||||||
|
|
||||||
|
# Good variable names regexes, separated by a comma. If names match any regex,
|
||||||
|
# they will always be accepted
|
||||||
|
good-names-rgxs=
|
||||||
|
|
||||||
|
# Include a hint for the correct naming format with invalid-name.
|
||||||
|
include-naming-hint=no
|
||||||
|
|
||||||
|
# Naming style matching correct inline iteration names.
|
||||||
|
inlinevar-naming-style=any
|
||||||
|
|
||||||
|
# Regular expression matching correct inline iteration names. Overrides
|
||||||
|
# inlinevar-naming-style.
|
||||||
|
#inlinevar-rgx=
|
||||||
|
|
||||||
|
# Naming style matching correct method names.
|
||||||
|
method-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct method names. Overrides method-naming-
|
||||||
|
# style.
|
||||||
|
#method-rgx=
|
||||||
|
|
||||||
|
# Naming style matching correct module names.
|
||||||
|
module-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct module names. Overrides module-naming-
|
||||||
|
# style.
|
||||||
|
#module-rgx=
|
||||||
|
|
||||||
|
# Colon-delimited sets of names that determine each other's naming style when
|
||||||
|
# the name regexes allow several styles.
|
||||||
|
name-group=
|
||||||
|
|
||||||
|
# Regular expression which should only match function or class names that do
|
||||||
|
# not require a docstring.
|
||||||
|
no-docstring-rgx=^_
|
||||||
|
|
||||||
|
# List of decorators that produce properties, such as abc.abstractproperty. Add
|
||||||
|
# to this list to register other decorators that produce valid properties.
|
||||||
|
# These decorators are taken in consideration only for invalid-name.
|
||||||
|
property-classes=abc.abstractproperty
|
||||||
|
|
||||||
|
# Naming style matching correct variable names.
|
||||||
|
variable-naming-style=snake_case
|
||||||
|
|
||||||
|
# Regular expression matching correct variable names. Overrides variable-
|
||||||
|
# naming-style.
|
||||||
|
#variable-rgx=
|
||||||
|
|
||||||
|
|
||||||
|
[FORMAT]
|
||||||
|
|
||||||
|
# Expected format of line ending, e.g. empty (any line ending), LF or CRLF.
|
||||||
|
expected-line-ending-format=
|
||||||
|
|
||||||
|
# Regexp for a line that is allowed to be longer than the limit.
|
||||||
|
ignore-long-lines=^\s*(# )?<?https?://\S+>?$
|
||||||
|
|
||||||
|
# Number of spaces of indent required inside a hanging or continued line.
|
||||||
|
indent-after-paren=4
|
||||||
|
|
||||||
|
# String used as indentation unit. This is usually " " (4 spaces) or "\t" (1
|
||||||
|
# tab).
|
||||||
|
indent-string=' '
|
||||||
|
|
||||||
|
# Maximum number of characters on a single line.
|
||||||
|
max-line-length=100
|
||||||
|
|
||||||
|
# Maximum number of lines in a module.
|
||||||
|
max-module-lines=1000
|
||||||
|
|
||||||
|
# List of optional constructs for which whitespace checking is disabled. `dict-
|
||||||
|
# separator` is used to allow tabulation in dicts, etc.: {1 : 1,\n222: 2}.
|
||||||
|
# `trailing-comma` allows a space between comma and closing bracket: (a, ).
|
||||||
|
# `empty-line` allows space-only lines.
|
||||||
|
no-space-check=trailing-comma,
|
||||||
|
dict-separator
|
||||||
|
|
||||||
|
# Allow the body of a class to be on the same line as the declaration if body
|
||||||
|
# contains single statement.
|
||||||
|
single-line-class-stmt=no
|
||||||
|
|
||||||
|
# Allow the body of an if to be on the same line as the test if there is no
|
||||||
|
# else.
|
||||||
|
single-line-if-stmt=no
|
||||||
|
|
||||||
|
|
||||||
|
[LOGGING]
|
||||||
|
|
||||||
|
# The type of string formatting that logging methods do. `old` means using %
|
||||||
|
# formatting, `new` is for `{}` formatting.
|
||||||
|
logging-format-style=old
|
||||||
|
|
||||||
|
# Logging modules to check that the string format arguments are in logging
|
||||||
|
# function parameter format.
|
||||||
|
logging-modules=logging
|
||||||
|
|
||||||
|
|
||||||
|
[MISCELLANEOUS]
|
||||||
|
|
||||||
|
# List of note tags to take in consideration, separated by a comma.
|
||||||
|
notes=FIXME,
|
||||||
|
XXX,
|
||||||
|
TODO
|
||||||
|
|
||||||
|
# Regular expression of note tags to take in consideration.
|
||||||
|
#notes-rgx=
|
||||||
|
|
||||||
|
|
||||||
|
[SIMILARITIES]
|
||||||
|
|
||||||
|
# Ignore comments when computing similarities.
|
||||||
|
ignore-comments=yes
|
||||||
|
|
||||||
|
# Ignore docstrings when computing similarities.
|
||||||
|
ignore-docstrings=yes
|
||||||
|
|
||||||
|
# Ignore imports when computing similarities.
|
||||||
|
ignore-imports=no
|
||||||
|
|
||||||
|
# Minimum lines number of a similarity.
|
||||||
|
min-similarity-lines=4
|
||||||
|
|
||||||
|
|
||||||
|
[SPELLING]
|
||||||
|
|
||||||
|
# Limits count of emitted suggestions for spelling mistakes.
|
||||||
|
max-spelling-suggestions=4
|
||||||
|
|
||||||
|
# Spelling dictionary name. Available dictionaries: none. To make it work,
|
||||||
|
# install the python-enchant package.
|
||||||
|
spelling-dict=
|
||||||
|
|
||||||
|
# List of comma separated words that should not be checked.
|
||||||
|
spelling-ignore-words=
|
||||||
|
|
||||||
|
# A path to a file that contains the private dictionary; one word per line.
|
||||||
|
spelling-private-dict-file=
|
||||||
|
|
||||||
|
# Tells whether to store unknown words to the private dictionary (see the
|
||||||
|
# --spelling-private-dict-file option) instead of raising a message.
|
||||||
|
spelling-store-unknown-words=no
|
||||||
|
|
||||||
|
|
||||||
|
[STRING]
|
||||||
|
|
||||||
|
# This flag controls whether inconsistent-quotes generates a warning when the
|
||||||
|
# character used as a quote delimiter is used inconsistently within a module.
|
||||||
|
check-quote-consistency=no
|
||||||
|
|
||||||
|
# This flag controls whether the implicit-str-concat should generate a warning
|
||||||
|
# on implicit string concatenation in sequences defined over several lines.
|
||||||
|
check-str-concat-over-line-jumps=no
|
||||||
|
|
||||||
|
|
||||||
|
[TYPECHECK]
|
||||||
|
|
||||||
|
# List of decorators that produce context managers, such as
|
||||||
|
# contextlib.contextmanager. Add to this list to register other decorators that
|
||||||
|
# produce valid context managers.
|
||||||
|
contextmanager-decorators=contextlib.contextmanager
|
||||||
|
|
||||||
|
# List of members which are set dynamically and missed by pylint inference
|
||||||
|
# system, and so shouldn't trigger E1101 when accessed. Python regular
|
||||||
|
# expressions are accepted.
|
||||||
|
generated-members=
|
||||||
|
|
||||||
|
# Tells whether missing members accessed in mixin class should be ignored. A
|
||||||
|
# mixin class is detected if its name ends with "mixin" (case insensitive).
|
||||||
|
ignore-mixin-members=yes
|
||||||
|
|
||||||
|
# Tells whether to warn about missing members when the owner of the attribute
|
||||||
|
# is inferred to be None.
|
||||||
|
ignore-none=yes
|
||||||
|
|
||||||
|
# This flag controls whether pylint should warn about no-member and similar
|
||||||
|
# checks whenever an opaque object is returned when inferring. The inference
|
||||||
|
# can return multiple potential results while evaluating a Python object, but
|
||||||
|
# some branches might not be evaluated, which results in partial inference. In
|
||||||
|
# that case, it might be useful to still emit no-member and other checks for
|
||||||
|
# the rest of the inferred objects.
|
||||||
|
ignore-on-opaque-inference=yes
|
||||||
|
|
||||||
|
# List of class names for which member attributes should not be checked (useful
|
||||||
|
# for classes with dynamically set attributes). This supports the use of
|
||||||
|
# qualified names.
|
||||||
|
ignored-classes=optparse.Values,thread._local,_thread._local
|
||||||
|
|
||||||
|
# List of module names for which member attributes should not be checked
|
||||||
|
# (useful for modules/projects where namespaces are manipulated during runtime
|
||||||
|
# and thus existing member attributes cannot be deduced by static analysis). It
|
||||||
|
# supports qualified module names, as well as Unix pattern matching.
|
||||||
|
ignored-modules=
|
||||||
|
|
||||||
|
# Show a hint with possible names when a member name was not found. The aspect
|
||||||
|
# of finding the hint is based on edit distance.
|
||||||
|
missing-member-hint=yes
|
||||||
|
|
||||||
|
# The minimum edit distance a name should have in order to be considered a
|
||||||
|
# similar match for a missing member name.
|
||||||
|
missing-member-hint-distance=1
|
||||||
|
|
||||||
|
# The total number of similar names that should be taken in consideration when
|
||||||
|
# showing a hint for a missing member.
|
||||||
|
missing-member-max-choices=1
|
||||||
|
|
||||||
|
# List of decorators that change the signature of a decorated function.
|
||||||
|
signature-mutators=
|
||||||
|
|
||||||
|
|
||||||
|
[VARIABLES]
|
||||||
|
|
||||||
|
# List of additional names supposed to be defined in builtins. Remember that
|
||||||
|
# you should avoid defining new builtins when possible.
|
||||||
|
additional-builtins=
|
||||||
|
|
||||||
|
# Tells whether unused global variables should be treated as a violation.
|
||||||
|
allow-global-unused-variables=yes
|
||||||
|
|
||||||
|
# List of strings which can identify a callback function by name. A callback
|
||||||
|
# name must start or end with one of those strings.
|
||||||
|
callbacks=cb_,
|
||||||
|
_cb
|
||||||
|
|
||||||
|
# A regular expression matching the name of dummy variables (i.e. expected to
|
||||||
|
# not be used).
|
||||||
|
dummy-variables-rgx=_+$|(_[a-zA-Z0-9_]*[a-zA-Z0-9]+?$)|dummy|^ignored_|^unused_
|
||||||
|
|
||||||
|
# Argument names that match this expression will be ignored. Default to name
|
||||||
|
# with leading underscore.
|
||||||
|
ignored-argument-names=_.*|^ignored_|^unused_
|
||||||
|
|
||||||
|
# Tells whether we should check for unused import in __init__ files.
|
||||||
|
init-import=no
|
||||||
|
|
||||||
|
# List of qualified module names which can have objects that can redefine
|
||||||
|
# builtins.
|
||||||
|
redefining-builtins-modules=six.moves,past.builtins,future.builtins,builtins,io
|
||||||
|
|
||||||
|
|
||||||
|
[CLASSES]
|
||||||
|
|
||||||
|
# List of method names used to declare (i.e. assign) instance attributes.
|
||||||
|
defining-attr-methods=__init__,
|
||||||
|
__new__,
|
||||||
|
setUp,
|
||||||
|
__post_init__
|
||||||
|
|
||||||
|
# List of member names, which should be excluded from the protected access
|
||||||
|
# warning.
|
||||||
|
exclude-protected=_asdict,
|
||||||
|
_fields,
|
||||||
|
_replace,
|
||||||
|
_source,
|
||||||
|
_make
|
||||||
|
|
||||||
|
# List of valid names for the first argument in a class method.
|
||||||
|
valid-classmethod-first-arg=cls
|
||||||
|
|
||||||
|
# List of valid names for the first argument in a metaclass class method.
|
||||||
|
valid-metaclass-classmethod-first-arg=cls
|
||||||
|
|
||||||
|
|
||||||
|
[DESIGN]
|
||||||
|
|
||||||
|
# Maximum number of arguments for function / method.
|
||||||
|
max-args=5
|
||||||
|
|
||||||
|
# Maximum number of attributes for a class (see R0902).
|
||||||
|
max-attributes=7
|
||||||
|
|
||||||
|
# Maximum number of boolean expressions in an if statement (see R0916).
|
||||||
|
max-bool-expr=5
|
||||||
|
|
||||||
|
# Maximum number of branch for function / method body.
|
||||||
|
max-branches=12
|
||||||
|
|
||||||
|
# Maximum number of locals for function / method body.
|
||||||
|
max-locals=15
|
||||||
|
|
||||||
|
# Maximum number of parents for a class (see R0901).
|
||||||
|
max-parents=7
|
||||||
|
|
||||||
|
# Maximum number of public methods for a class (see R0904).
|
||||||
|
max-public-methods=20
|
||||||
|
|
||||||
|
# Maximum number of return / yield for function / method body.
|
||||||
|
max-returns=6
|
||||||
|
|
||||||
|
# Maximum number of statements in function / method body.
|
||||||
|
max-statements=50
|
||||||
|
|
||||||
|
# Minimum number of public methods for a class (see R0903).
|
||||||
|
min-public-methods=2
|
||||||
|
|
||||||
|
|
||||||
|
[IMPORTS]
|
||||||
|
|
||||||
|
# List of modules that can be imported at any level, not just the top level
|
||||||
|
# one.
|
||||||
|
allow-any-import-level=
|
||||||
|
|
||||||
|
# Allow wildcard imports from modules that define __all__.
|
||||||
|
allow-wildcard-with-all=no
|
||||||
|
|
||||||
|
# Analyse import fallback blocks. This can be used to support both Python 2 and
|
||||||
|
# 3 compatible code, which means that the block might have code that exists
|
||||||
|
# only in one or another interpreter, leading to false positives when analysed.
|
||||||
|
analyse-fallback-blocks=no
|
||||||
|
|
||||||
|
# Deprecated modules which should not be used, separated by a comma.
|
||||||
|
deprecated-modules=optparse,tkinter.tix
|
||||||
|
|
||||||
|
# Create a graph of external dependencies in the given file (report RP0402 must
|
||||||
|
# not be disabled).
|
||||||
|
ext-import-graph=
|
||||||
|
|
||||||
|
# Create a graph of every (i.e. internal and external) dependencies in the
|
||||||
|
# given file (report RP0402 must not be disabled).
|
||||||
|
import-graph=
|
||||||
|
|
||||||
|
# Create a graph of internal dependencies in the given file (report RP0402 must
|
||||||
|
# not be disabled).
|
||||||
|
int-import-graph=
|
||||||
|
|
||||||
|
# Force import order to recognize a module as part of the standard
|
||||||
|
# compatibility libraries.
|
||||||
|
known-standard-library=
|
||||||
|
|
||||||
|
# Force import order to recognize a module as part of a third party library.
|
||||||
|
known-third-party=enchant
|
||||||
|
|
||||||
|
# Couples of modules and preferred modules, separated by a comma.
|
||||||
|
preferred-modules=
|
||||||
|
|
||||||
|
|
||||||
|
[EXCEPTIONS]
|
||||||
|
|
||||||
|
# Exceptions that will emit a warning when being caught. Defaults to
|
||||||
|
# "BaseException, Exception".
|
||||||
|
overgeneral-exceptions=BaseException,
|
||||||
|
Exception
|
@ -0,0 +1,16 @@
|
|||||||
|
'''
|
||||||
|
This file is part of JANINE.
|
||||||
|
|
||||||
|
JANINE is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
JANINE is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with JANINE. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
'''
|
265
janine/janine.py
265
janine/janine.py
@ -1,11 +1,25 @@
|
|||||||
|
'''
|
||||||
|
This file is part of JANINE.
|
||||||
|
|
||||||
|
JANINE is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
JANINE is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with JANINE. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
'''
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
import datetime
|
|
||||||
import configparser
|
import configparser
|
||||||
import threading
|
|
||||||
import logging
|
import logging
|
||||||
import signal
|
|
||||||
import asyncio
|
import asyncio
|
||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
|
|
||||||
@ -14,23 +28,23 @@ from aioxmpp.structs import PresenceShow
|
|||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from janine.utils import find_one, find_all
|
from janine.utils import find_one, make_msg, format_warning
|
||||||
from janine.sources import sources_from_config, MiscDataSources
|
from janine.sources import sources_from_config, MiscDataSources
|
||||||
|
|
||||||
log = logging.getLogger('janine')
|
log = logging.getLogger('janine')
|
||||||
log.setLevel(logging.INFO)
|
log.setLevel(logging.INFO)
|
||||||
log.addHandler(logging.StreamHandler())
|
log.addHandler(logging.StreamHandler())
|
||||||
|
|
||||||
Warning = namedtuple('Warning', ['id',
|
Warning_ = namedtuple('Warning_', ['id',
|
||||||
'sent',
|
'sent',
|
||||||
'effective_from',
|
'effective_from',
|
||||||
'expires',
|
'expires',
|
||||||
'urgency',
|
'urgency',
|
||||||
'sender',
|
'sender',
|
||||||
'headline',
|
'headline',
|
||||||
'description',
|
'description',
|
||||||
'instruction',
|
'instruction',
|
||||||
'landkreise'])
|
'landkreise'])
|
||||||
|
|
||||||
|
|
||||||
def stub_warning(id_):
|
def stub_warning(id_):
|
||||||
@ -38,35 +52,35 @@ def stub_warning(id_):
|
|||||||
Returns a stubbed warning for loading warnings from disk.
|
Returns a stubbed warning for loading warnings from disk.
|
||||||
The only real attribute is the @id_ .
|
The only real attribute is the @id_ .
|
||||||
'''
|
'''
|
||||||
return Warning(id=id_,
|
return Warning_(id=id_,
|
||||||
sent='',
|
sent='',
|
||||||
effective_from='',
|
effective_from='',
|
||||||
expires='',
|
expires='',
|
||||||
urgency='',
|
urgency='',
|
||||||
sender='',
|
sender='',
|
||||||
headline='',
|
headline='',
|
||||||
description='',
|
description='',
|
||||||
instruction='',
|
instruction='',
|
||||||
landkreise=[])
|
landkreise=[])
|
||||||
|
|
||||||
def to_warning(data):
|
def to_warning(data):
|
||||||
'''
|
'''
|
||||||
Returns a Warning given the raw data
|
Returns a Warning given the raw data
|
||||||
'''
|
'''
|
||||||
info = find_one(lambda x: 'headline' in x.keys(), data['info'])
|
info = find_one(lambda x: 'headline' in x.keys(), data['info'])
|
||||||
return Warning(id=data['identifier'],
|
return Warning_(id=data['identifier'],
|
||||||
sent=data['sent'],
|
sent=data['sent'],
|
||||||
# Not all items have to have those
|
# Not all items have to have those
|
||||||
effective_from=info.get('effective', 'N/A'),
|
effective_from=info.get('effective', 'N/A'),
|
||||||
# Not all items have to have those
|
# Not all items have to have those
|
||||||
expires=info.get('expires', 'N/A'),
|
expires=info.get('expires', 'N/A'),
|
||||||
urgency=info['urgency'],
|
urgency=info['urgency'],
|
||||||
# Not all items have to have those
|
# Not all items have to have those
|
||||||
sender=info.get('senderName', 'N/A'),
|
sender=info.get('senderName', 'N/A'),
|
||||||
headline=info['headline'],
|
headline=info['headline'],
|
||||||
description=info['description'],
|
description=info['description'],
|
||||||
instruction=info.get('instruction', 'N/A'),
|
instruction=info.get('instruction', 'N/A'),
|
||||||
landkreise=get_landkreise(data))
|
landkreise=get_landkreise(data))
|
||||||
|
|
||||||
def get_landkreise(data):
|
def get_landkreise(data):
|
||||||
'''
|
'''
|
||||||
@ -76,7 +90,7 @@ def get_landkreise(data):
|
|||||||
geocode = find_one(lambda e: 'geocode' in e.keys(), info['area'])
|
geocode = find_one(lambda e: 'geocode' in e.keys(), info['area'])
|
||||||
|
|
||||||
# Note: Some items may have multiple Landkreise
|
# Note: Some items may have multiple Landkreise
|
||||||
return list(map(lambda e: e['valueName'], geocode['geocode']))
|
return list(map(lambda e: e['valueName'], geocode['geocode']))
|
||||||
|
|
||||||
def parse_data(text):
|
def parse_data(text):
|
||||||
'''
|
'''
|
||||||
@ -86,6 +100,10 @@ def parse_data(text):
|
|||||||
return [to_warning(el) for el in data]
|
return [to_warning(el) for el in data]
|
||||||
|
|
||||||
class WarningBot:
|
class WarningBot:
|
||||||
|
'''
|
||||||
|
This class represents the actual bot. The only things
|
||||||
|
to be done is call connect() after creating an instance.
|
||||||
|
'''
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._warnings0 = []
|
self._warnings0 = []
|
||||||
self._warnings1 = []
|
self._warnings1 = []
|
||||||
@ -105,28 +123,28 @@ class WarningBot:
|
|||||||
Starts the "event loop" of the bot
|
Starts the "event loop" of the bot
|
||||||
'''
|
'''
|
||||||
self._client = aioxmpp.PresenceManagedClient(
|
self._client = aioxmpp.PresenceManagedClient(
|
||||||
self._jid,
|
self._jid,
|
||||||
aioxmpp.make_security_layer(self._password))
|
aioxmpp.make_security_layer(self._password))
|
||||||
|
|
||||||
async with self._client.connected() as stream:
|
async with self._client.connected():
|
||||||
log.info('Client connected to server')
|
log.info('Client connected to server')
|
||||||
|
|
||||||
# In case you want a nice avatar
|
# In case you want a nice avatar
|
||||||
if self._avatar:
|
if self._avatar:
|
||||||
log.info('Setting avatar')
|
log.info('Setting avatar')
|
||||||
with open(self._avatar, 'rb') as f:
|
with open(self._avatar, 'rb') as avatar_file:
|
||||||
image_data = f.read()
|
image_data = avatar_file.read()
|
||||||
|
|
||||||
avatar_set = aioxmpp.avatar.AvatarSet()
|
avatar_set = aioxmpp.avatar.AvatarSet()
|
||||||
avatar_set.add_avatar_image('image/png', image_bytes=image_data)
|
avatar_set.add_avatar_image('image/png', image_bytes=image_data)
|
||||||
await (self._client.summon(aioxmpp.avatar.AvatarService)
|
await (self._client.summon(aioxmpp.avatar.AvatarService)
|
||||||
.publish_avatar_set(avatar_set))
|
.publish_avatar_set(avatar_set))
|
||||||
|
|
||||||
# Set some presence information
|
# Set some presence information
|
||||||
self._client.set_presence(
|
self._client.set_presence(
|
||||||
aioxmpp.PresenceState(available=True,
|
aioxmpp.PresenceState(available=True,
|
||||||
show=PresenceShow.CHAT),
|
show=PresenceShow.CHAT),
|
||||||
self._status)
|
self._status)
|
||||||
|
|
||||||
# Enable Carbons
|
# Enable Carbons
|
||||||
await self._client.summon(aioxmpp.CarbonsClient).enable()
|
await self._client.summon(aioxmpp.CarbonsClient).enable()
|
||||||
@ -134,9 +152,9 @@ class WarningBot:
|
|||||||
|
|
||||||
# Register the message handler
|
# Register the message handler
|
||||||
self._client.stream.register_message_callback(
|
self._client.stream.register_message_callback(
|
||||||
aioxmpp.MessageType.CHAT,
|
aioxmpp.MessageType.CHAT,
|
||||||
None,
|
None,
|
||||||
self._handle_message)
|
self._handle_message)
|
||||||
log.info('Message handler registered')
|
log.info('Message handler registered')
|
||||||
|
|
||||||
# Start our fetch-send loop
|
# Start our fetch-send loop
|
||||||
@ -145,7 +163,7 @@ class WarningBot:
|
|||||||
# use async in event handlers
|
# use async in event handlers
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
periodic = loop.create_task(self._periodic_requests())
|
periodic = loop.create_task(self._periodic_requests())
|
||||||
log.info('Periodic ticker started')
|
log.info('Periodic ticker started')
|
||||||
await periodic
|
await periodic
|
||||||
|
|
||||||
def __is_message_valid(self, msg):
|
def __is_message_valid(self, msg):
|
||||||
@ -154,25 +172,14 @@ class WarningBot:
|
|||||||
'''
|
'''
|
||||||
if msg.type_ != aioxmpp.MessageType.CHAT:
|
if msg.type_ != aioxmpp.MessageType.CHAT:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if not msg.body:
|
if not msg.body:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if msg.from_.domain != self._jid.domain and self._same_domain:
|
if msg.from_.domain != self._jid.domain and self._same_domain:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def __make_msg(self, to, body):
|
|
||||||
'''
|
|
||||||
Wrapper for creating a message object to enqueue or send.
|
|
||||||
'''
|
|
||||||
msg = aioxmpp.Message(
|
|
||||||
type_=aioxmpp.MessageType.CHAT,
|
|
||||||
to=to)
|
|
||||||
msg.body[None] = body
|
|
||||||
|
|
||||||
return msg
|
|
||||||
|
|
||||||
def _handle_message(self, msg):
|
def _handle_message(self, msg):
|
||||||
# Handle cases we don't want to deal with
|
# Handle cases we don't want to deal with
|
||||||
@ -185,74 +192,78 @@ class WarningBot:
|
|||||||
if cmd == 'subscribe':
|
if cmd == 'subscribe':
|
||||||
# Do we have a landkreis?
|
# Do we have a landkreis?
|
||||||
if len(cmd_parts) < 2:
|
if len(cmd_parts) < 2:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body='Du hast keinen Landkreis angegeben'))
|
body='Du hast keinen Landkreis angegeben'))
|
||||||
return
|
return
|
||||||
|
|
||||||
# Check if the entered Landkreis is valid
|
# Check if the entered Landkreis is valid
|
||||||
landkreis = ' '.join(cmd_parts[1:])
|
landkreis = ' '.join(cmd_parts[1:])
|
||||||
if not landkreis in self._channels:
|
if not landkreis in self._channels:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body='Der angegebene Landkreis ist ungültig'))
|
body='Der angegebene Landkreis ist ungültig'))
|
||||||
return
|
return
|
||||||
|
|
||||||
if not landkreis in self._warn_clients.keys():
|
if landkreis not in self._warn_clients.keys():
|
||||||
self._warn_clients[landkreis] = []
|
self._warn_clients[landkreis] = []
|
||||||
self._warn_clients[landkreis].append(str(msg.from_.bare()))
|
self._warn_clients[landkreis].append(str(msg.from_.bare()))
|
||||||
|
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=f'Du erhälst nun Nachrichten zu {landkreis} von mir'))
|
body=f'Du erhälst nun Nachrichten zu {landkreis} von mir'))
|
||||||
|
|
||||||
with open(self._client_store, 'w') as cf:
|
with open(self._client_store, 'w') as clients_file:
|
||||||
cf.write(json.dumps(self._warn_clients))
|
clients_file.write(json.dumps(self._warn_clients))
|
||||||
|
|
||||||
# Send all known warnings for the landkreis to the user
|
# Send all known warnings for the landkreis to the user
|
||||||
for warning in self._warnings0:
|
for warning in self._warnings0:
|
||||||
if landkreis in warning.landkreise:
|
if landkreis in warning.landkreise:
|
||||||
body = self._format_warning(warning)
|
body = format_warning(warning)
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=body))
|
body=body))
|
||||||
|
|
||||||
elif cmd == 'unsubscribe':
|
elif cmd == 'unsubscribe':
|
||||||
# Do we have a landkreis?
|
# Do we have a landkreis?
|
||||||
if len(cmd_parts) < 2:
|
if len(cmd_parts) < 2:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body='Du hast keinen Landkreis angegeben'))
|
body='Du hast keinen Landkreis angegeben'))
|
||||||
return
|
return
|
||||||
|
|
||||||
landkreis = ' '.join(cmd_parts[1:])
|
landkreis = ' '.join(cmd_parts[1:])
|
||||||
if not landkreis in self._warn_clients:
|
if landkreis not in self._warn_clients:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=f'Du hast {landkreis} nicht abonniert'))
|
body=f'Du hast {landkreis} nicht abonniert'))
|
||||||
return
|
return
|
||||||
|
|
||||||
if str(msg.from_.bare()) in self._warn_clients[landkreis]:
|
if str(msg.from_.bare()) in self._warn_clients[landkreis]:
|
||||||
filter_ = lambda x: x != str(msg.from_.bare())
|
filter_ = lambda x: x != str(msg.from_.bare())
|
||||||
self._warn_clients[landkreis] = list(filter(filter_,
|
self._warn_clients[landkreis] = list(filter(filter_,
|
||||||
self._warn_clients[landkreis]))
|
self._warn_clients[landkreis]))
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=f'Du erhälst keine Nachrichten zu {landkreis} mehr von mir'))
|
body=f'Du erhälst keine Nachrichten zu {landkreis} mehr von mir'))
|
||||||
|
|
||||||
if len(self._warn_clients[landkreis]) == 0:
|
if len(self._warn_clients[landkreis]) == 0:
|
||||||
del self._warn_clients[landkreis]
|
del self._warn_clients[landkreis]
|
||||||
else:
|
else:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=f'Du hast {landkreis} nicht abonniert'))
|
body=f'Du hast {landkreis} nicht abonniert'))
|
||||||
elif cmd == 'help':
|
elif cmd == 'help':
|
||||||
body = 'Verfügbare Befehle:\n\nsubscribe <Landkreis> - Abonniere einen Landkreis\nunsubscribe <Landkreis> - Entferne das Abonnement zu einem Landkreis\nhelp - Gebe diese Hilfe aus'
|
body = '''Verfügbare Befehle:
|
||||||
self._client.enqueue(self.__make_msg(
|
|
||||||
|
subscribe <Landkreis> - Abonniere einen Landkreis
|
||||||
|
unsubscribe <Landkreis> - Entferne das Abonnement zu einem Landkreis
|
||||||
|
help - Gebe diese Hilfe aus'''
|
||||||
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body=body))
|
body=body))
|
||||||
else:
|
else:
|
||||||
self._client.enqueue(self.__make_msg(
|
self._client.enqueue(make_msg(
|
||||||
to=msg.from_,
|
to=msg.from_,
|
||||||
body='Diesen Befehl kenne ich nicht... Mit "help" kannst du alle Befehle sehen, die ich kenne.'))
|
body='Diesen Befehl kenne ich nicht... Mit "help" kannst du alle Befehle sehen, die ich kenne.'))
|
||||||
|
|
||||||
@ -265,11 +276,11 @@ class WarningBot:
|
|||||||
while True:
|
while True:
|
||||||
log.debug('Refreshing warning list')
|
log.debug('Refreshing warning list')
|
||||||
await self._fetch_warnings()
|
await self._fetch_warnings()
|
||||||
|
|
||||||
# Flush the warnings to disk
|
# Flush the warnings to disk
|
||||||
ids = list(map(lambda x: x.id, self._warnings0))
|
ids = list(map(lambda x: x.id, self._warnings0))
|
||||||
with open(self._warnings_file, 'w') as wf:
|
with open(self._warnings_file, 'w') as warnings_file:
|
||||||
wf.write(json.dumps(ids))
|
warnings_file.write(json.dumps(ids))
|
||||||
|
|
||||||
await asyncio.sleep(self._refresh_timeout)
|
await asyncio.sleep(self._refresh_timeout)
|
||||||
|
|
||||||
@ -283,15 +294,15 @@ class WarningBot:
|
|||||||
for source in self._sources:
|
for source in self._sources:
|
||||||
req = requests.get(source)
|
req = requests.get(source)
|
||||||
self._warnings0 = parse_data(req.text)
|
self._warnings0 = parse_data(req.text)
|
||||||
|
|
||||||
# Find new warnings and send the new ones
|
# Find new warnings and send the new ones
|
||||||
ids = map(lambda x: x.id, self._warnings1)
|
ids = map(lambda x: x.id, self._warnings1)
|
||||||
for warning in self._warnings0:
|
for warning in self._warnings0:
|
||||||
if warning.id in ids:
|
if warning.id in ids:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if len(set(warning.landkreise).intersection(self._warn_clients.keys())):
|
if len(set(warning.landkreise).intersection(self._warn_clients.keys())):
|
||||||
body = self._format_warning(warning)
|
body = format_warning(warning)
|
||||||
for landkreis in warning.landkreise:
|
for landkreis in warning.landkreise:
|
||||||
for to in self._warn_clients.get(landkreis, []):
|
for to in self._warn_clients.get(landkreis, []):
|
||||||
msg = aioxmpp.stanza.Message(
|
msg = aioxmpp.stanza.Message(
|
||||||
@ -300,87 +311,54 @@ class WarningBot:
|
|||||||
msg.body[None] = body
|
msg.body[None] = body
|
||||||
await self._client.send(msg)
|
await self._client.send(msg)
|
||||||
|
|
||||||
def __time_format(self, time_str):
|
|
||||||
'''
|
|
||||||
Reformat ISO style time data to a more
|
|
||||||
readable format.
|
|
||||||
'''
|
|
||||||
date = None
|
|
||||||
try:
|
|
||||||
date = datetime.datetime.fromisoformat(time_str)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if not date:
|
|
||||||
return time_str
|
|
||||||
|
|
||||||
return '{}.{}.{} {}:{}'.format(date.day,
|
|
||||||
date.month,
|
|
||||||
date.year,
|
|
||||||
date.hour,
|
|
||||||
date.minute)
|
|
||||||
|
|
||||||
def _format_warning(self, warning):
|
|
||||||
'''
|
|
||||||
Send a warning to all the recipients
|
|
||||||
'''
|
|
||||||
# Reformat the message a bit
|
|
||||||
effective_time = self.__time_format(warning.effective_from)
|
|
||||||
expiry_time = self.__time_format(warning.expires)
|
|
||||||
body = f'*{warning.headline}*\n({effective_time} bis {expiry_time})\n\n{warning.description}'
|
|
||||||
|
|
||||||
# Smells like script injection, but okay
|
|
||||||
body = body.replace('<br>', '\n')
|
|
||||||
body = body.replace('<br/>', '\n')
|
|
||||||
return body
|
|
||||||
|
|
||||||
def _load_config(self):
|
def _load_config(self):
|
||||||
# Load config
|
# Load config
|
||||||
config_path = sys.argv[1] if len(sys.argv) == 2 else '/etc/janine/janine.conf'
|
config_path = sys.argv[1] if len(sys.argv) == 2 else '/etc/janine/janine.conf'
|
||||||
config = configparser.ConfigParser()
|
config = configparser.ConfigParser()
|
||||||
config.read(config_path)
|
config.read(config_path)
|
||||||
|
|
||||||
# Configure sources
|
# Configure sources
|
||||||
self._sources = sources_from_config(config)
|
self._sources = sources_from_config(config)
|
||||||
self._data_dir = config['General'].get('DataDir', '/etc/janine')
|
self._data_dir = config['General'].get('DataDir', '/etc/janine')
|
||||||
self._recipients = config['General']['Recipients'].split(',')
|
self._recipients = config['General']['Recipients'].split(',')
|
||||||
self._refresh_timeout = int(config['General']['Timeout'])
|
self._refresh_timeout = int(config['General']['Timeout'])
|
||||||
self._same_domain = config['General'].get('SameDomain', 'True') == 'True'
|
self._same_domain = config['General'].get('SameDomain', 'True') == 'True'
|
||||||
|
|
||||||
# Persistent data
|
# Persistent data
|
||||||
# Subscribed clients
|
# Subscribed clients
|
||||||
self._client_store = os.path.join(self._data_dir, 'clients.json')
|
self._client_store = os.path.join(self._data_dir, 'clients.json')
|
||||||
if os.path.exists(self._client_store):
|
if os.path.exists(self._client_store):
|
||||||
with open(self._client_store, 'r') as cf:
|
with open(self._client_store, 'r') as clients_file:
|
||||||
self._warn_clients = json.loads(cf.read())
|
self._warn_clients = json.loads(clients_file.read())
|
||||||
log.info('Clients read from disk')
|
log.info('Clients read from disk')
|
||||||
|
|
||||||
self._warnings_file = os.path.join(self._data_dir, 'warnings.json')
|
self._warnings_file = os.path.join(self._data_dir, 'warnings.json')
|
||||||
if os.path.exists(self._warnings_file):
|
if os.path.exists(self._warnings_file):
|
||||||
with open(self._warnings_file, 'r') as wf:
|
with open(self._warnings_file, 'r') as warnings_file:
|
||||||
self._warnings0 = list(map(stub_warning, json.loads(wf.read())))
|
self._warnings0 = list(map(stub_warning,
|
||||||
|
json.loads(warnings_file.read())))
|
||||||
log.info('Warnings read from disk')
|
log.info('Warnings read from disk')
|
||||||
|
|
||||||
# Landkreise
|
# Landkreise
|
||||||
self._channels = []
|
self._channels = []
|
||||||
channels = {}
|
channels = {}
|
||||||
channel_file = os.path.join(self._data_dir, 'channels.json')
|
self._channel_file = os.path.join(self._data_dir, 'channels.json')
|
||||||
if not os.path.exists(channel_file):
|
if not os.path.exists(self._channel_file):
|
||||||
log.info('Requesting search channels')
|
log.info('Requesting search channels')
|
||||||
req = requests.get(MiscDataSources.channels())
|
req = requests.get(MiscDataSources.channels())
|
||||||
channels = json.loads(req.text)
|
channels = json.loads(req.text)
|
||||||
self._channels = list(map(lambda key: channels[key].get('NAME', ''),
|
self._channels = list(map(lambda key: channels[key].get('NAME', ''),
|
||||||
channels.keys()))
|
channels.keys()))
|
||||||
|
|
||||||
try:
|
try:
|
||||||
with open(channel_file, 'w') as f:
|
with open(self._channel_file, 'w') as channel_file:
|
||||||
f.write(json.dumps(self._channels))
|
channel_file.write(json.dumps(self._channels))
|
||||||
except Exception as err:
|
except Exception as err:
|
||||||
log.error('Failed to cache channel data:')
|
log.error('Failed to cache channel data:')
|
||||||
log.error(str(err))
|
log.error(str(err))
|
||||||
else:
|
else:
|
||||||
with open(channel_file, 'r') as f:
|
with open(self._channel_file, 'r') as channel_file:
|
||||||
self._channels = json.loads(f.read())
|
self._channels = json.loads(channel_file.read())
|
||||||
log.info('Search channels read from disk')
|
log.info('Search channels read from disk')
|
||||||
|
|
||||||
# Bot Config
|
# Bot Config
|
||||||
@ -390,6 +368,9 @@ class WarningBot:
|
|||||||
self._status = config['Bot'].get('Status', 'Warnt dich vor Katastrophen')
|
self._status = config['Bot'].get('Status', 'Warnt dich vor Katastrophen')
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
'''
|
||||||
|
Main function.
|
||||||
|
'''
|
||||||
bot = WarningBot()
|
bot = WarningBot()
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
loop.run_until_complete(bot.connect())
|
loop.run_until_complete(bot.connect())
|
||||||
|
@ -1,3 +1,20 @@
|
|||||||
|
'''
|
||||||
|
This file is part of JANINE.
|
||||||
|
|
||||||
|
JANINE is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
JANINE is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with JANINE. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
'''
|
||||||
|
|
||||||
class WarningSources:
|
class WarningSources:
|
||||||
'''
|
'''
|
||||||
A collection of sources of the BBK
|
A collection of sources of the BBK
|
||||||
|
@ -1,3 +1,64 @@
|
|||||||
|
'''
|
||||||
|
This file is part of JANINE.
|
||||||
|
|
||||||
|
JANINE is free software: you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
JANINE is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with JANINE. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
'''
|
||||||
|
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
import aioxmpp
|
||||||
|
|
||||||
|
def make_msg(self, to, body):
|
||||||
|
'''
|
||||||
|
Wrapper for creating a message object to enqueue or send.
|
||||||
|
'''
|
||||||
|
msg = aioxmpp.Message(
|
||||||
|
type_=aioxmpp.MessageType.CHAT,
|
||||||
|
to=to)
|
||||||
|
msg.body[None] = body
|
||||||
|
|
||||||
|
return msg
|
||||||
|
|
||||||
|
def format_time(self, time_str):
|
||||||
|
'''
|
||||||
|
Reformat ISO style time data to a more
|
||||||
|
readable format.
|
||||||
|
'''
|
||||||
|
try:
|
||||||
|
date = datetime.datetime.fromisoformat(time_str)
|
||||||
|
except ValueError:
|
||||||
|
return time_str
|
||||||
|
|
||||||
|
return f'{date.day}.{date.month}.{date.year} {date.hour}:{date.minute}'
|
||||||
|
|
||||||
|
def format_warning(self, warning):
|
||||||
|
'''
|
||||||
|
Send a warning to all the recipients
|
||||||
|
'''
|
||||||
|
# Reformat the message a bit
|
||||||
|
effective_time = format_time(warning.effective_from)
|
||||||
|
expiry_time = format_time(warning.expires)
|
||||||
|
body = f'''*{warning.headline}*
|
||||||
|
({effective_time} bis {expiry_time})
|
||||||
|
|
||||||
|
{warning.description}'''
|
||||||
|
|
||||||
|
# Smells like script injection, but okay
|
||||||
|
body = body.replace('<br>', '\n')
|
||||||
|
body = body.replace('<br/>', '\n')
|
||||||
|
return body
|
||||||
|
|
||||||
def find_one(func, array):
|
def find_one(func, array):
|
||||||
'''
|
'''
|
||||||
Utility function
|
Utility function
|
||||||
|
Loading…
Reference in New Issue
Block a user