| 10 | from config.data import logging |
| 11 | |
| 12 | class Output: |
| 13 | def __init__(self): |
| 14 | self.nowTime = time.strftime("%Y%m%d%H%M%S",time.localtime()) |
| 15 | Webinfo.result = Webinfo.result + Urlerror.result |
| 16 | self.filename_json = self.nowTime + '.json' |
| 17 | self.filename_xls = self.nowTime + '.xlsx' |
| 18 | self.path_json = os.path.join(path.output,self.filename_json) |
| 19 | self.path_xls = os.path.join(path.output,self.filename_xls) |
| 20 | if Save.format == 'json' and Webinfo.result: |
| 21 | self.outJson() |
| 22 | if Save.format == 'xlsx' and Webinfo.result: |
| 23 | self.outXls() |
| 24 | |
| 25 | |
| 26 | def outJson(self): |
| 27 | with open(self.path_json,'w') as file: |
| 28 | file.write(json.dumps(Webinfo.result)) |
| 29 | print() |
| 30 | successMsg = "结果文件输出路径为:{0}".format(self.path_json) |
| 31 | logging.success(successMsg) |
| 32 | |
| 33 | def outXls(self): |
| 34 | with xlsxwriter.Workbook(self.path_xls) as workbook: |
| 35 | worksheet = workbook.add_worksheet('Finger scan') |
| 36 | bold = workbook.add_format({"bold":True,"valign":"center"}) |
| 37 | worksheet.set_column('A:A', 30) |
| 38 | worksheet.set_column('B:B', 40) |
| 39 | worksheet.set_column('C:C', 30) |
| 40 | worksheet.set_column('D:D', 20) |
| 41 | worksheet.set_column('E:E', 15) |
| 42 | worksheet.set_column('F:F', 15) |
| 43 | worksheet.set_column('G:G', 30) |
| 44 | worksheet.set_column('H:H', 10) |
| 45 | worksheet.set_column('I:I', 30) |
| 46 | worksheet.set_column('J:J', 30) |
| 47 | worksheet.write('A1', 'Url', bold) |
| 48 | worksheet.write('B1', 'Title', bold) |
| 49 | worksheet.write('C1', 'cms', bold) |
| 50 | worksheet.write('D1', 'Server', bold) |
| 51 | worksheet.write('E1', 'status', bold) |
| 52 | worksheet.write('F1', 'size', bold) |
| 53 | worksheet.write('G1', 'ip', bold) |
| 54 | worksheet.write('H1', 'iscdn', bold) |
| 55 | worksheet.write('I1', 'address', bold) |
| 56 | worksheet.write('J1', 'isp', bold) |
| 57 | row = 1 |
| 58 | col = 0 |
| 59 | for vaule in Webinfo.result: |
| 60 | worksheet.write(row, col, vaule["url"]) |
| 61 | worksheet.write(row, col+1, vaule["title"]) |
| 62 | worksheet.write(row, col+2, vaule["cms"]) |
| 63 | worksheet.write(row, col+3, vaule["Server"]) |
| 64 | worksheet.write(row, col+4, vaule["status"]) |
| 65 | worksheet.write(row, col+5, vaule["size"]) |
| 66 | worksheet.write(row, col+6, vaule["ip"]) |
| 67 | worksheet.write(row, col+7, vaule["iscdn"]) |
| 68 | worksheet.write(row, col+8, vaule["address"]) |
| 69 | worksheet.write(row, col+9, vaule["isp"]) |