Add python script, cli and guit
This commit is contained in:
parent
051f4b3002
commit
1bef81de3e
GPG key ID:
9570A7DB7CB2F436
41
cli.py
Normal file
41
cli.py
Normal file
|
|
@ -0,0 +1,41 @@
|
|||
##########
|
||||
# IMPORT #
|
||||
##########
|
||||
import argparse
|
||||
from os.path import exists, isfile
|
||||
from parser import ParserJuniper
|
||||
|
||||
__author__ = "rick@gnous.eu"
|
||||
__licence__ = "GPL3"
|
||||
|
||||
parserJuniper = ParserJuniper()
|
||||
parserArg = argparse.ArgumentParser(
|
||||
description="Parse a Juniper conf file and print the series of set commands for."
|
||||
)
|
||||
parserArg.add_argument("-f", "--file", nargs=1, required=True, type=str, help="The conf file.")
|
||||
parserArg.add_argument("-o", "--output", nargs=1, type=str, help="The output file.")
|
||||
|
||||
arg = parserArg.parse_args()
|
||||
|
||||
inputFile = str(arg.file[0])
|
||||
outputFile = None
|
||||
if arg.output:
|
||||
outputFile = str(arg.output[0])
|
||||
if exists(outputFile) and isfile(outputFile):
|
||||
writeOverFile = input("The file already exists, write over it ? o/O")
|
||||
if writeOverFile.lower() != 'o':
|
||||
print("STOP EVERYTHING!!!!!!!!!!!!!")
|
||||
exit(0)
|
||||
elif exists(outputFile):
|
||||
print("The output musts be a file.")
|
||||
exit(0)
|
||||
|
||||
if exists(inputFile) and isfile(inputFile):
|
||||
parseResul = parserJuniper.parseFile(inputFile)
|
||||
if outputFile:
|
||||
with open(outputFile, 'w') as file:
|
||||
file.write(parseResul)
|
||||
else:
|
||||
print(parseResul)
|
||||
else:
|
||||
print("Pass an existing file.")
|
||||
36
controllers/controller.py
Normal file
36
controllers/controller.py
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
##########
|
||||
# IMPORT #
|
||||
##########
|
||||
from PyQt5.QtWidgets import QTextEdit, QTextBrowser
|
||||
from parser import ParserJuniper
|
||||
|
||||
__author__ = "rick@gnous.eu"
|
||||
__licence__ = "GPL3"
|
||||
|
||||
class Controller:
|
||||
def __init__(self, inputText, outputText):
|
||||
"""
|
||||
Init the controller
|
||||
|
||||
:param inputText QTextEdit: the area where Juniper conf in write
|
||||
:param outputText QTextBrowser: area where a series of set
|
||||
command is showed
|
||||
"""
|
||||
self.inputText = inputText
|
||||
self.outputText = outputText
|
||||
self.parser = ParserJuniper()
|
||||
|
||||
def click(self):
|
||||
"""
|
||||
Called when the user press the Parse button.
|
||||
Gets the text of inputText and parse it. Shows the result on
|
||||
outputText.
|
||||
"""
|
||||
self.parser.resetTree()
|
||||
textToParse = self.inputText.toPlainText()
|
||||
parsedText = ""
|
||||
for line in textToParse.splitlines():
|
||||
textConf = self.parser.parse(line)
|
||||
if textConf:
|
||||
parsedText += textConf + "\n"
|
||||
self.outputText.setText(parsedText)
|
||||
36
gui.py
Normal file
36
gui.py
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
##########
|
||||
# IMPORT #
|
||||
##########
|
||||
from sys import argv
|
||||
from PyQt5 import uic
|
||||
from PyQt5.QtWidgets import QMainWindow, QApplication, QPushButton, QTextEdit,\
|
||||
QTextBrowser
|
||||
from controllers.controller import Controller
|
||||
|
||||
__author__ = "rick@gnous.eu"
|
||||
__licence__ = "GPL3"
|
||||
|
||||
class Interface(QMainWindow):
|
||||
def __init__(self):
|
||||
super(Interface, self).__init__()
|
||||
|
||||
self.setWindowTitle("Parser Juniper")
|
||||
self.ui = uic.loadUi("views/principal.ui")
|
||||
self.setCentralWidget(self.ui)
|
||||
|
||||
inputText = self.ui.findChildren(QTextEdit, "inputText")[0]
|
||||
outputText = self.ui.findChildren(QTextBrowser, "outputText")[0]
|
||||
parseButton = self.ui.findChildren(QPushButton, "parse")[0]
|
||||
quitButton = self.ui.findChildren(QPushButton, "quit")[0]
|
||||
|
||||
self.controller = Controller(inputText, outputText)
|
||||
|
||||
parseButton.clicked.connect(self.controller.click)
|
||||
quitButton.clicked.connect(quit)
|
||||
|
||||
self.show()
|
||||
|
||||
if __name__ == "__main__":
|
||||
app = QApplication(argv)
|
||||
window = Interface()
|
||||
exit(app.exec_())
|
||||
|
|
@ -1,85 +0,0 @@
|
|||
#!/usr/bin/php
|
||||
<?php
|
||||
|
||||
/******************************************************************************************************
|
||||
*
|
||||
* Written by Tim Price 24/11/2016
|
||||
*
|
||||
* The purpose of this script is to take a juniper config and convert it to set commands.
|
||||
* There was a business need to audit thbe correctness of Juniper configs and the process of logging
|
||||
* into each Juniper device and dumping the configs in set format was too burdensome on both the Juniper
|
||||
* devices in question but also extended the execution time of the audit scripts by some minutes.
|
||||
*
|
||||
* Usage example: `cat juniper-config.txt | juniper-config-to-set.php`
|
||||
*
|
||||
******************************************************************************************************/
|
||||
|
||||
function endswith($string, $test) {
|
||||
$strlen = strlen($string);
|
||||
$testlen = strlen($test);
|
||||
if ($testlen > $strlen) return false;
|
||||
return substr_compare($string, $test, $strlen - $testlen, $testlen) === 0;
|
||||
}
|
||||
|
||||
function addtotree($tree,$word) {
|
||||
array_push($tree,$word);
|
||||
return $tree;
|
||||
}
|
||||
|
||||
function removewordfromtree($tree) {
|
||||
array_pop($tree);
|
||||
return $tree;
|
||||
}
|
||||
|
||||
function printtree($tree) {
|
||||
return implode(" ",$tree);
|
||||
}
|
||||
|
||||
$tree=array();
|
||||
|
||||
while($line = fgets(STDIN)){
|
||||
|
||||
// Skip any commented lines
|
||||
if(strpos($line,'#')===0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Trim white space from the line
|
||||
$line=trim($line);
|
||||
|
||||
// Trim inline comments from the line
|
||||
if(preg_match('/; ##/',$line)==1) {
|
||||
list($line,$comment)=preg_split('/; ##/',$line);
|
||||
$line .= ";";
|
||||
}
|
||||
|
||||
// What remains is a series of tests to see what the line ends with.
|
||||
|
||||
// This test matches a ';' character which means its a leaf and we can output the whole line to STDOUT
|
||||
if(endswith($line,';')) {
|
||||
|
||||
// Trim the semi-colon from the end of the line, we don't need that.
|
||||
$line = rtrim($line,';');
|
||||
|
||||
//Test to see if the tree is empty, if it is then we'll just print the leaf and not the tree
|
||||
//to avoid excess white spaces
|
||||
if(count($tree)==0) {
|
||||
echo "set $line\n";
|
||||
} else {
|
||||
echo "set " . printtree($tree) . " $line\n";
|
||||
}
|
||||
}
|
||||
|
||||
// This test matches a '{' character on the end of the line and means we need to add a branch to the tree
|
||||
if(endswith($line,'{')) {
|
||||
$line = rtrim($line,' {');
|
||||
$tree=addtotree($tree,$line);
|
||||
}
|
||||
|
||||
// This test matches a '}' character on the end of the line and means we need to remove the last branch that we added to the tree
|
||||
if(endswith($line,'}')) {
|
||||
$tree=removewordfromtree($tree);
|
||||
}
|
||||
}
|
||||
|
||||
?>
|
||||
64
parser.py
Normal file
64
parser.py
Normal file
|
|
@ -0,0 +1,64 @@
|
|||
# Tom Price writes the program in PHP here : https://github.com/pgnuta/juniper-config-to-set
|
||||
# I adapt it in Python and add some enhancements
|
||||
__author__ = "Tim Price | rick@gnous.eu"
|
||||
__licence__ = "GPL3"
|
||||
|
||||
class ParserJuniper:
|
||||
def __init__(self):
|
||||
self.tree = []
|
||||
|
||||
def resetTree(self):
|
||||
self.tree = []
|
||||
|
||||
def printTree(self, tree):
|
||||
return ''.join(map(str, tree))
|
||||
|
||||
def parse(self, line):
|
||||
"""
|
||||
Parse a line of conf
|
||||
|
||||
:param line str: line will be parse
|
||||
:ret str: a parse string, empty if its a comment
|
||||
"""
|
||||
ret = ""
|
||||
line = line.strip()
|
||||
if not line.startswith('#'):
|
||||
if '#' in line:
|
||||
line, comment = line.split('#', 1)
|
||||
line = line.strip()
|
||||
|
||||
if line.endswith(';'):
|
||||
line = line[:-1]
|
||||
|
||||
if not self.tree:
|
||||
ret = "set " + line
|
||||
else:
|
||||
ret = "set " + self.printTree(self.tree) + line
|
||||
|
||||
if line.endswith('{'):
|
||||
line = line[:-1]
|
||||
self.tree.append(line)
|
||||
|
||||
if line.endswith('}'):
|
||||
self.tree.pop()
|
||||
|
||||
return ret
|
||||
|
||||
def parseFile(self, path):
|
||||
"""
|
||||
parse a file and return the commands
|
||||
|
||||
:param path str: the path to file
|
||||
:ret str: the series of set commands
|
||||
"""
|
||||
ret = ""
|
||||
with open(path, 'r') as file:
|
||||
self.resetTree()
|
||||
for line in file:
|
||||
lineConf = self.parse(line)
|
||||
if lineConf:
|
||||
ret += lineConf + "\n"
|
||||
return ret
|
||||
|
||||
#parser = ParserJuniper()
|
||||
#parser.parseFile("test")
|
||||
78
views/principal.ui
Normal file
78
views/principal.ui
Normal file
|
|
@ -0,0 +1,78 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<ui version="4.0">
|
||||
<class>MainWindow</class>
|
||||
<widget class="QMainWindow" name="MainWindow">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>0</x>
|
||||
<y>0</y>
|
||||
<width>800</width>
|
||||
<height>550</height>
|
||||
</rect>
|
||||
</property>
|
||||
<property name="windowTitle">
|
||||
<string>MainWindow</string>
|
||||
</property>
|
||||
<widget class="QWidget" name="centralwidget">
|
||||
<widget class="QTextEdit" name="inputText">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>20</x>
|
||||
<y>20</y>
|
||||
<width>361</width>
|
||||
<height>391</height>
|
||||
</rect>
|
||||
</property>
|
||||
</widget>
|
||||
<widget class="QPushButton" name="parse">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>420</x>
|
||||
<y>430</y>
|
||||
<width>131</width>
|
||||
<height>41</height>
|
||||
</rect>
|
||||
</property>
|
||||
<property name="text">
|
||||
<string>Parse !</string>
|
||||
</property>
|
||||
</widget>
|
||||
<widget class="QTextBrowser" name="outputText">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>420</x>
|
||||
<y>20</y>
|
||||
<width>351</width>
|
||||
<height>391</height>
|
||||
</rect>
|
||||
</property>
|
||||
</widget>
|
||||
<widget class="QPushButton" name="quit">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>250</x>
|
||||
<y>430</y>
|
||||
<width>131</width>
|
||||
<height>41</height>
|
||||
</rect>
|
||||
</property>
|
||||
<property name="text">
|
||||
<string>Quit</string>
|
||||
</property>
|
||||
</widget>
|
||||
</widget>
|
||||
<widget class="QMenuBar" name="menubar">
|
||||
<property name="geometry">
|
||||
<rect>
|
||||
<x>0</x>
|
||||
<y>0</y>
|
||||
<width>800</width>
|
||||
<height>20</height>
|
||||
</rect>
|
||||
</property>
|
||||
</widget>
|
||||
<widget class="QStatusBar" name="statusbar"/>
|
||||
</widget>
|
||||
<resources/>
|
||||
<connections/>
|
||||
</ui>
|
||||
Loading…
Reference in a new issue