rw_method.py

import csv
import numpy as np


class QA:

    def __init__(self, filename):
        self.answers = []
        self.questions = []
        self.index = []
        self.read_csv = None
        self.filename = filename
        self.load()

    def add(self, question, answer):
        self.index = np.append(self.index, str(int(self.index[-1]) + 1))
        self.questions = np.append(self.questions, question)
        self.answers = np.append(self.answers, answer)

    def edit(self, index, question, answer):
        v = np.where(self.index == index)[0]
        self.questions[v] = question
        self.answers[v] = answer

    def delete(self, index):
        v = np.where(self.index == index)[0]
        self.index = np.delete(self.index, v)
        self.questions = np.delete(self.questions, v)
        self.answers = np.delete(self.answers, v)

    def renumber(self):
        m = 0
        while self.index[m] == '0':
            m += 1
        self.index[m:] = np.array([str(i) for i in range(m, len(self.index) + 1)])

    def list_copy(self):
        return self.index[4:].copy(), self.questions[4:].copy(), self.answers[4:].copy()

    def load(self):
        with open(self.filename) as qa:
            self.read_csv = np.array(list(csv.reader(qa)))
            self.index = self.read_csv[1:, 0]
            self.questions = self.read_csv[1:, 1]
            self.answers = self.read_csv[1:, 2]

    def save(self):
        np.savetxt(self.filename, np.array([self.index, self.questions, self.answers]).T, delimiter=",",
                   header='序号,问,答',
                   comments="", encoding='utf-8', fmt='%s')


class User:
    def __init__(self, filename):
        self.filename = filename
        self.index = None
        self.users = None
        self.permissions = None
        self.read_csv = None
        self.load()

    def load(self):
        with open(self.filename) as user:
            self.read_csv = np.array(list(csv.reader(user)))
            self.index = self.read_csv[1:, 0]
            self.users = self.read_csv[1:, 1]
            self.permissions = self.read_csv[1:, 2]

    def get_permission(self, user):
        if user in self.users:
            return self.permissions[self.users == user]
        else:
            return '0'

    def change_permission(self, user, permission):
        if user not in self.users:
            self.index = np.append(self.index, str(int(self.index[-1]) + 1))
            self.users = np.append(self.users, user)
            self.permissions = np.append(self.permissions, permission)
        else:
            self.permissions[self.users == user] = permission