def write_payload(self):
if not os.path.exists('./output'):
os.makedirs('./output')
with open(r'./output/' + self.payload_name, 'w') as f:
if "go_" in self.payload_name and self.cleanup:
print("[*] Removing Comments and Print Statements")
self.payload_output = removeCommentsGo(self.payload_output,removePrint=True)
elif "python_" in self.payload_name and self.cleanup:
print("[*] Removing Comments and Print Statements")
self.payload_output = removeCommentsPy(self.payload_output,removePrint=True)
elif self.payload_name and self.cleanup:
print("[!] Error Selecting Type of File for Cleaning : %s" % (self.payload_name))
f.write(self.payload_output)
def write_payload(self):
if not os.path.exists('./output'):
os.makedirs('./output')
with open(r'./output/' + self.payload_name, 'w') as f:
if "go_" in self.payload_name and self.cleanup:
print "[*] Removing Comments and Print Statements"
self.payload_output = removeCommentsGo(self.payload_output,removePrint=True)
elif "python_" in self.payload_name and self.cleanup:
print "[*] Removing Comments and Print Statements"
self.payload_output = removeCommentsPy(self.payload_output,removePrint=True)
elif self.payload_name and self.cleanup:
print "[!] Error Selecting Type of File for Cleaning : %s" % (self.payload_name)
f.write(self.payload_output)