如何使python脚本正常工作以从.txt文件中获取输入并返回肯定数

Question

I have a python script that is supposed to take a directory full of .txt files and determine if each .txt file return positive or negative for matching certain text statements inside the file itself like "known infection source". However, my script doesn't work and returns the following error message. Any help would be greatly appreciated!

Sample JSON file text

{
    "detected_referrer_samples": [
        {
            "positives": 1,
            "sha256": "325f928105efb4c227be1a83fb3d0634ec5903bdfce2c3580ad113fc0f15373c",
            "total": 52
        },
        {
            "positives": 20,
            "sha256": "48d85943ea9cdd1e480d73556e94d8438c1b2a8a30238dff2c52dd7f5c047435",
            "total": 53
        }
    ],
    "detected_urls": [],
    "domain_siblings": [],
    "resolutions": [],
    "response_code": 1,
    "verbose_msg": "Domain found in dataset",
    "whois": null
}

Error

Traceback (most recent call last):
  File "vt_reporter1.py", line 35, in <module>
    print(vt_result_check(path))
  File "vt_reporter1.py", line 20, in vt_result_check
    vt_result |= any(sample['positives'] > 0 for sample_type in sample_types
  File "vt_reporter1.py", line 21, in <genexpr>
    for sample in vt_data.get(sample_type, []))
AttributeError: 'list' object has no attribute 'get'

Code

import os
import json
import csv

path=r'./output/'
csvpath='C:/Users/bwerner/Documents'

def vt_result_check(path):
    vt_result = False
    for filename in os.listdir(path):
        with open(path + filename, 'r') as vt_result_file:
            vt_data = json.load(vt_result_file)

        # Look for any positive detected referrer samples
        # Look for any positive detected communicating samples
        # Look for any positive detected downloaded samples
        # Look for any positive detected URLs
        sample_types = ('detected_referrer_samples', 'detected_communicating_samples',
                        'detected_downloaded_samples', 'detected_urls')
        vt_result |= any(sample['positives'] > 0 for sample_type in sample_types
                                                 for sample in vt_data.get(sample_type, []))

        # Look for a Dr. Web category of known infection source
        vt_result |= vt_data.get('Dr.Web category') == "known infection source"

        # Look for a Forecepoint ThreatSeeker category of elevated exposure
        # Look for a Forecepoint ThreatSeeker category of phishing and other frauds
        # Look for a Forecepoint ThreatSeeker category of suspicious content
        threats = ("elevated exposure", "phishing and other frauds", "suspicious content")
        vt_result |= vt_data.get('Forcepoint ThreatSeeker category') in threats

    return vt_result

if __name__ == '__main__':
    print(vt_result_check(path))
    with open(csvpath, 'w') as csvfile:
        writer.writerow([vt_result_check(path)])

Answer 1

The error tells you everything you need to know about what is going wrong, which is that you cannot call the get() function on a list. In Python, the get() function can only be used with dictionaries, which are different from lists. Instead of using the get() function, call a specific index of the list and your program should work. For example:

for sample in list[10:11]

which returns the 11th element of the list.

Answer 2

Can you post the contents of the file, or some text that represents the content of the file that is being read in?

Here's some feedback based on what is seen in the code you posted:

1. vt_result_file contains valid JSON

2. And whatever it is reading, is being read into python as a List. We can determine this because of the error that you are receiving. Look at the last line of the error:

   AttributeError: 'list' object has no attribute 'get'

   It says that you are trying to access the "get" attribute on a "list". Looking at your code, we can see that you are calling "get" on "vt_data" three times:

   • Once as

      vt_data.get(sample_type, [])  

   • Another time as

      vt_data.get('Dr.Web category')  

   • And finally as

      vt_data.get('Forcepoint ThreatSeeker category')  

   Per the error message, your variable vt_data is a list and not a dictionary.

So you need to ask yourself:

Were you expecting vt_result_file to contain a dictionary? If so, open the file and examine what is contained there, and turn it into a dictionary.

Unfortunately, without seeing the contents of this file, it is hard to suggest what you need to change to fix this error.