|
5 | 5 | measurement_name = (sys.argv[5]) # get measurement from argv |
6 | 6 | print ('Measurement-name: '+measurement_name) |
7 | 7 |
|
8 | | -# argv1 = outsideip, agrv2 = Domain, argv3 length, argv4 tragetip |
| 8 | +# argv1 = outsideip, agrv2 = Domain, argv3 length, argv4 tragetip, sys.argv[5] bucketname, sys.argv[6] date, sys.argv[7] asn |
| 9 | + |
| 10 | + |
| 11 | +asn = str(sys.argv[7]) |
9 | 12 |
|
10 | 13 | import geoip2.database |
11 | 14 | import socket |
|
27 | 30 | Target = str(sys.argv[4]) |
28 | 31 | reader.close() |
29 | 32 |
|
30 | | -reader = geoip2.database.Reader('/geolite/GeoLite2-ASN.mmdb') |
31 | | -response = reader.asn(str(sys.argv[1])) |
32 | | -Asn = response.autonomous_system_organization |
33 | | -reader.close() |
| 33 | +if asn =='true': |
| 34 | + reader = geoip2.database.Reader('/geolite/GeoLite2-ASN.mmdb') |
| 35 | + response = reader.asn(str(sys.argv[1])) |
| 36 | + Asn = response.autonomous_system_organization |
| 37 | + reader.close() |
34 | 38 |
|
35 | 39 | # print to log |
36 | 40 | print (Country) |
37 | 41 | print (State) |
38 | 42 | print (City) |
39 | 43 | print (Zip) |
40 | | -print (Asn) |
41 | 44 | print (Long) |
42 | 45 | print (Lat) |
43 | 46 | print (ISO) |
| 47 | +if asn =='true': |
| 48 | + print (Asn) |
44 | 49 | print ('Outside IP: ', IP) |
45 | 50 | print ('Target IP: ', Target) |
46 | 51 | print ('Domain: ', Domain) |
|
110 | 115 | point.tag("Name", Country) |
111 | 116 | point.tag("IP", IP), |
112 | 117 | point.tag("Target", Target) |
113 | | -point.tag("Asn", Asn) |
| 118 | +if asn =='true': |
| 119 | + point.tag("Asn", Asn) |
114 | 120 |
|
115 | 121 | point.field("Domain", Domain) |
116 | 122 | point.field("latitude", Lat) |
|
120 | 126 | point.field("key", ISO) |
121 | 127 | point.field("IP", IP) |
122 | 128 | point.field("Target", Target) |
123 | | -point.field("Asn", Asn) |
| 129 | +if asn =='true': |
| 130 | + point.field("Asn", Asn) |
124 | 131 | point.field("Name", Country) |
125 | 132 | point.field("duration", duration) |
126 | 133 | point.field("metric", 1) |
|
0 commit comments