mirror of https://github.com/secdev/scapy.git
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
#! /usr/bin/env python
|
|
|
|
"Process python files to improve python 3 migration"
|
|
|
|
from __future__ import print_function
|
|
import os
|
|
import sys
|
|
import getopt
|
|
# Modified glob version to support **
|
|
import glob2
|
|
import re
|
|
|
|
def main():
|
|
tabsize = 8
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], "t:")
|
|
if not args:
|
|
raise getopt.error, "At least one file argument required"
|
|
except getopt.error as msg:
|
|
print(msg)
|
|
print("usage:", sys.argv[0], "files ...")
|
|
return
|
|
|
|
files = []
|
|
for arg in args:
|
|
files.extend(glob2.glob(arg))
|
|
|
|
for filename in files:
|
|
if "autoFixer" in filename or "build" in filename:
|
|
continue
|
|
process(filename, tabsize)
|
|
|
|
# Utils, regexes
|
|
r1_ = r'([( ,=])(?<![b])\"(([^\n\\\"]|\\.)*(\\x|\\0)([^\n\\\"]|\\.)*)\"'
|
|
r2_ = r"([( ,=])(?<![b])\'(([^\n\\\']|\\.)*(\\x|\\0)([^\n\\\']|\\.)*)\'"
|
|
#r3_ = r"(?<![\"'\\])(['\"])\1([^\"'])"
|
|
|
|
r1_r = r'\g<1>b"\2"'
|
|
r2_r = r"\g<1>b'\2'"
|
|
#r3_r = r"b\1\1\2"
|
|
|
|
def process(filename, tabsize):
|
|
try:
|
|
f = open(filename)
|
|
text = f.read()
|
|
f.close()
|
|
except IOError as msg:
|
|
print("%r: I/O error: %s" % (filename, msg))
|
|
return
|
|
# Remove tabs
|
|
newtext = text.expandtabs(tabsize)
|
|
# Auto-detect bytes with "\x...", "\0..."
|
|
newtext = re.sub(r1_, r1_r, newtext)
|
|
# Auto-detect bytes with '\x...', '\0...'
|
|
newtext = re.sub(r2_, r2_r, newtext)
|
|
# Auto-detect bytes with '', "" but not """, '''
|
|
#newtext = re.sub(r3_, r3_r, newtext)
|
|
if newtext == text:
|
|
return
|
|
f = open(filename, "w")
|
|
f.write(newtext)
|
|
f.close()
|
|
print(filename)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|