96 lines
2.7 KiB
Python
Executable File
96 lines
2.7 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
"""Python REGEX. A standalone tool by Evan Pratten <evan@ewpratten.com>
|
|
|
|
usage: py-re [-h] [-g GROUPS] [-A] [-I] [-M] [-S] [-X] expression [file]
|
|
|
|
A tool for REGEX using Python's re module.
|
|
|
|
positional arguments:
|
|
expression Regular Expression
|
|
file File to search (default: stdin)
|
|
|
|
options:
|
|
-h, --help show this help message and exit
|
|
-g GROUPS, --groups GROUPS
|
|
Print these matching groups (comma-separated)
|
|
-A, --ascii ASCII-only matching
|
|
-I, --ignore-case Case-insensitive matching
|
|
-M, --multiline Multiline matching
|
|
-S, --dotall Dot matches all
|
|
-X, --verbose Verbose regex
|
|
"""
|
|
|
|
import argparse
|
|
import sys
|
|
import re
|
|
from pathlib import Path
|
|
|
|
|
|
def main() -> int:
|
|
# Handle program arguments
|
|
ap = argparse.ArgumentParser(
|
|
prog="py-re", description="A tool for REGEX using Python's re module."
|
|
)
|
|
ap.add_argument("expression", help="Regular Expression")
|
|
ap.add_argument(
|
|
"file",
|
|
help="File to search (default: stdin)",
|
|
default=None,
|
|
nargs="?",
|
|
type=Path,
|
|
)
|
|
ap.add_argument(
|
|
"-g",
|
|
"--groups",
|
|
help="Print these matching groups (comma-separated)",
|
|
default=None,
|
|
type=lambda x: [int(i) for i in x.split(",")],
|
|
)
|
|
|
|
# RE flags
|
|
ap.add_argument("-A", "--ascii", help="ASCII-only matching", action="store_true")
|
|
ap.add_argument(
|
|
"-I", "--ignore-case", help="Case-insensitive matching", action="store_true"
|
|
)
|
|
ap.add_argument("-M", "--multiline", help="Multiline matching", action="store_true")
|
|
ap.add_argument("-S", "--dotall", help="Dot matches all", action="store_true")
|
|
ap.add_argument("-X", "--verbose", help="Verbose regex", action="store_true")
|
|
args = ap.parse_args()
|
|
# args.groups = args.groups[0] if args.groups else None
|
|
|
|
# Build the regex flags
|
|
re_flags = re.NOFLAG
|
|
if args.ascii:
|
|
re_flags |= re.ASCII
|
|
if args.ignore_case:
|
|
re_flags |= re.IGNORECASE
|
|
if args.multiline:
|
|
re_flags |= re.MULTILINE
|
|
if args.dotall:
|
|
re_flags |= re.DOTALL
|
|
if args.verbose:
|
|
re_flags |= re.VERBOSE
|
|
|
|
# Compile the regular expression
|
|
regex = re.compile(args.expression)
|
|
|
|
# Read the file
|
|
file = args.file.open() if args.file else sys.stdin
|
|
|
|
# Search the file
|
|
for line in file:
|
|
match = regex.search(line)
|
|
if match:
|
|
if args.groups:
|
|
for group in args.groups:
|
|
print(match.group(group), end=" ")
|
|
else:
|
|
print(match.group(0), end=" ")
|
|
print()
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|