Запись в файл CSV? - PullRequest
Новая
       5

Запись в файл CSV?

0 голосов
/ 12 марта 2020

Я довольно новичок в python, но я написал этот код, чтобы прочитать файл CSV, пинговать список IP-адресов в первом столбце и выводить статус и IP-адрес в другой файл CSV, используя функцию записи CSV. В конце концов я хочу иметь возможность записать только статус в новый CSV-файл, потому что все остальные данные не изменятся. Может ли кто-нибудь помочь мне сделать это? По сути, мне нужен способ записи только в указанный столбец c в TEST2.csv 

import platform    # For getting the operating system name
import subprocess as sp  # For executing a shell command
import csv

def ping(host):
    """
    Returns True if host (str) responds to a ping request.
    Remember that a host may not respond to a ping (ICMP) request even if the host 
    name is valid.
    """

    # Option for the number of packets as a function of
    param = '-n' if platform.system().lower()=='windows' else '-c'

    # Building the command. Ex: "ping -c 1 google.com"
    command = ['ping', param, '2', host]

    return sp.call(command) == 0

with open('TEST.csv') as rfile:
    reader = csv.reader(rfile)
    count = 0 
    status = 0
    strstatus = ''
    with open('TEST2.csv','w',newline='') as wfile:
        fieldnames = ['a','b','c','d','IP Address', 'Status']
        # a,b,c,d are placeholders for other other fieldnames in datafile
        writer = csv.DictWriter(wfile,fieldnames=fieldnames)
        next(reader)
        writer.writeheader()
        for IP in reader:
            status = ping(IP)
            if status:
                strstatus = 'Online'
            else:
                strstatus = 'Offline'
            writer.writerow({'a':None, 'b':None, 'c':None , 'd':None , 'IP Address' : 
IP,'Status' : strstatus})
            count += 1
            if count > 4:
                break
rfile.close()
wfile.close()

1 Ответ

0 голосов
/ 12 марта 2020

Вид полного переписывания с использованием некоторого вашего кода, но он должен выполнять свою работу: 

import platform
import subprocess as sp
import csv

INPUT_FILENAME = 'Input.csv'
INPUT_FIELDNAMES = ['column_1', 'column_2', 'ip']
INPUT_DELIMITER = ';'

OUTPUT_FILENAME = 'Output.csv'
OUTPUT_DELIMITER = ';'
OUTPUT_FIELDNAMES = ['column_1', 'column_2', 'ip', 'status']

NUMBER_OF_PINGS = 2
pings = str(NUMBER_OF_PINGS)


def ping(host, pings=pings):
    # https://stackoverflow.com/a/32684938/3991125
    param = '-n' if platform.system().lower() == 'windows' else '-c'
    command = ['ping', param, pings, host]
    print('Pinging {}'.format(host))
    is_available = sp.call(command, stdout=sp.DEVNULL, stderr=sp.DEVNULL) == 0
    print('--> available: {}'.format(is_available))
    return is_available


def read_csv(filepath, fieldnames, delimiter):
    with open(filepath) as input_file:
        reader = csv.DictReader(input_file, fieldnames=fieldnames, delimiter=delimiter)
        data = [row for row in reader]
    return data


def work(data):
    for d in data:
        ip = d.get('ip')
        is_available = ping(ip)
        d['status'] = is_available
    return data


def write_csv(data, filepath, fieldnames, delimiter):
    with open(filepath, 'w') as output_file:
        writer = csv.DictWriter(output_file, fieldnames=fieldnames, delimiter=delimiter)
        writer.writeheader()
        writer.writerows(data)


if __name__ == "__main__":
    data = read_csv(INPUT_FILENAME, INPUT_FIELDNAMES, INPUT_DELIMITER)
    processed = work(data)
    write_csv(processed, OUTPUT_FILENAME, OUTPUT_FIELDNAMES, OUTPUT_DELIMITER)
...