Question

我的csv文件的内容如下

101,item_1 101,item_1

如果是csv，我的以下代码将执行

import csv    
fName = input()
def read_csv(fName):
    try:
        with open(fName, 'r') as f:
            reader = csv.reader(f)
            for row in reader:
                print (row)

read_csv(fName)

这里介绍了如何在decorator函数中编写异常并在其顶部调用。

第一位装饰师

如果fName不以.txt或.csv结尾，则它必须生成输出not accept

第二个装饰器

如果fName = file.txt文本文件，则必须注意以下操作

def read_txt(fName):
    f = open(fName, "r")
    print(f.readline())

如果使用csv，则执行第一个功能，如果使用txt，则执行下一个功能。如何实现使用装饰器。我可以提出条件来解决问题，但事实并非如此

下面没有装饰器的整个代码

fName = input()
def read_csv(fName):
    if fName.endswith('.csv'):
        #print  ('hi')
        try:
            with open(fName, 'r') as f:
                reader = csv.reader(f)
                for row in reader:
                    print (row)
        except IOError:
            print ("Could not read file:", fName)
    #SECOND DECORATOR
    if fName.endswith('.txt'):
        f = open(fName, "r")
        print(f.readline())
    #FIRST DECORATOR
    if not(fName.endswith('.csv')) and not(fName.endswith('.txt')):
        print ('not accept')
read_csv(fName)

Answer 1

您的问题似乎不是在装饰器下，而是在工厂模式下，即根据输入文件进行不同的处理。

下面的代码是一个非常简单且基本的Factory模式解决方案，可以根据您的需要进行相应的修改，

import os
from abc import ABC, abstractmethod


class FileProcessor(ABC):
    @abstractmethod
    def process():
        pass

class TextFileProcessor(FileProcessor):
    def process(self, file_path):
        print("Text file processing goes here")


class CsvFileProcessor(FileProcessor):
    def process(self, file_path):
        print("CSV file processing goes here")


class DefaultFileProcessor(FileProcessor):
    def process(self, file_path):
        raise ValueError("File %s is not valid" % file_path)


class FileFactory:
    processors = {
        'txt': TextFileProcessor,
        'csv': CsvFileProcessor,
        'default': DefaultFileProcessor
    }

    def __init__(self, file_path):
        if not os.path.exists(file_path):
            raise IOError("File not found")
        self.file_path = file_path

    def process(self):
        dot_splits = self.file_path.split(".")
        ext = dot_splits[-1] if len(dot_splits) > 1 else "default"
        ext = ext if ext in self.processors else "default"
        processor_class = self.processors.get(ext)

        return processor_class().process(self.file_path)


FileFactory(file_path).process()

在以后的阶段，如果您想添加json处理器，则可以通过添加

轻松完成

processors = {
    'txt': TextFileProcessor,
    'csv': CsvFileProcessor,
    'json': JsonFileProcessor,
    'default': DefaultFileProcessor
}

并创建新的Json处理器类，

class JsonFileProcessor(FileProcessor):
    def process(self, file_path):
        print("JSON file processing goes here")

Answer 2

您可以使用装饰器来做到这一点：

import functools


def check_arguments(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        fname = kwargs['fname']
        if not fname.endswith('.csv') and not fname.endswith('.txt'):
            print('not accept')
        return func(*args, **kwargs)
    return wrapper


def set_file_processor(func):
    def read_csv(fname):
        print('read_csv', fname)

    def read_txt(fname):
        print('read_txt', fname)


    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        fname = kwargs['fname']
        if fname.endswith('.csv'):
            read_csv(fname)
        elif fname.endswith('.txt'):
            read_txt(fname)
        return func(*args, **kwargs)
    return wrapper


@check_arguments
@set_file_processor
def process(fname):
    pass


process(fname='input.csv')

Answer 3

根据您的代码和this very useful guide，以下是可能的解决方案：

def read_file_decorator(fName):
    def read_csv():
        print('read_csv')
        with open(fName, 'r') as f:
            reader = csv.reader(f)
            for row in reader:
                print(row)

    def read_txt():
        print('read_txt')
        f = open(fName, 'r')
        for row in f:
            print(row)

    if fName.endswith('.csv'):
        return read_csv
    elif fName.endswith('.txt'):
        return read_txt
    else:
        return None

reader_function = read_file_decorator(fileName)
if reader_function != None:
    reader_function()
else:
    print('not accept')

我使用一个有状态的装饰器，在实际执行读取器功能之前记住该文件名（以免传递两次）；我将固定值None用于无效的文件类型。

Answer 4

根据要求，使用装饰器会过分夸大装饰器。但是，如果必须使用装饰器实现此功能，则可以通过以下方式实现：

我们可以创建一个名为read_file的虚拟函数和一个名为reader的装饰器函数
用户将始终以文件名作为参数调用read_file，装饰器函数reader将检查传递的文件扩展名并调用所需的函数-read_csv或read_text

def reader(fun):
    def wrapper(*args):
        fname = args[0]
        if fname.endswith('.csv'):
            read_csv(fname)
        elif fname.endswith('.txt'):
            read_text(fname)
        else:
            print('not accepted')
    return wrapper

def read_csv(fname):
    print('In read_csv()')

def read_text(fname):
    print('In read_text()')

@reader
def read_file(fname):
    pass


read_file('a.csv')
read_file('a.txt')
read_file('filename.py')

输出

In read_csv()
In read_text()
not accepted

如何在文件操作中编写两个装饰器

4 个答案: