Lint only *.py files with --diff option

When using the `--diff` option flake8 only lints *.py files or files
defined by the `--filename` filter.

Fixes #362