[edu] 연락처 (push, pull, csv)

[연락처(common dict) ]

폰트 사이즈 12 고딕

폰트 사이즈 12 고딕

폰트 사이즈 12 고딕

폰트 사이즈 12 고딕

폰트 사이즈 12 고딕

class Contact:
    tokens = []
    file = "out.csv"

    @classmethod
    def set_tokens(cls, *args): cls.tokens = [arg for arg in args]

    @classmethod
    def set_file(cls, file): cls.file = file

    def __init__(self): self.members = list()

    @classmethod
    def to_member(cls, **kwargs):
        info = {}
        for token in Contact.tokens:
            info[token] = kwargs[token] if token in kwargs else None
        return info

    def indexes(self, **kwargs):
        r = []
        info = Contact.to_member(**kwargs)
        for i, member in enumerate(self.members):
            is_member = True
            for key in info.keys():
                if info[key] and info[key] != member[key]:
                    is_member = False
                    break
            if is_member:
                r.append(i)
        return r

    def filter(self, **kwargs): return [self.members[i] for i in range(len(self.members)) if i in self.indexes(**kwargs)]

    def show(self, **kwargs):
        r = list(map(str, self.filter(**kwargs)))
        if not len(r):
            return "empty"
        return "\n".join(r)

    def add(self, **kwargs): self.members.append(self.to_member(**kwargs))

    def remove(self, **kwargs):
        r = self.indexes(**kwargs)
        for i in range(len(r)-1, -1, -1):
            del self.members[r[i]]

    @classmethod
    def to_str(cls, member):
        r = ""
        for token in Contact.tokens:
            r += str(member[token]) + ","
        return r[:-1]

    def push(self):
        with open(Contact.file, "w") as f:
            f.write(",".join(Contact.tokens)+"\n")
            for member in self.members:
                f.write(Contact.to_str(member)+"\n")

    @classmethod
    def is_tokens(cls, line):
        values = line[:-1].split(",")
        if len(values) != len(Contact.tokens):
            return False
        for i, token in enumerate(Contact.tokens):
            if token != values[i]:
                return False
        return True

    def pull(self):
        r = []
        with open(Contact.file, "r") as f:
            if not Contact.is_tokens(f.readline()):
                return
            while True:
                line = f.readline()
                if not line:
                    break
                values = line[:-1].split(",")
                info = {}
                for i, token in enumerate(Contact.tokens):
                    info[token] = values[i]
                r.append(info)
        self.members = r

    def operation(self, cmd):
        cmd = cmd.split(",")
        if len(cmd) and cmd[0] in self.add.__name__:
            info = {}
            values = cmd[1:]
            if len(values) != len(Contact.tokens):
                print("conflict")
                return
            for i, token in enumerate(Contact.tokens):
                info[token] = values[i] if values[i] != " " else None
            self.add(**info)

    def __str__(self): return self.show()


c = Contact()
Contact.set_tokens("name", "phone", "addr", "email")

while True:
    c.operation(input())
    print(c.show())

# c.add(name="kim", phone="01022223333", addr="서울시", email="contact@gmail.com")
# c.add(addr="서울시", email="contact@gmail.com")
# c.add(addr="서울시", email="contact@gmail.com")
# print(c.filter(name="kim", addr="서울시"))
# c.pull()
# print(c)
# c.remove(name="kim")
# print(c)