#!/usr/bin/env python3 import ListUsers import csv import io import configparser import lib.uis.default as default_cmd # Follows -u, -a, -f flags class Backup: """Backups a Tilde database to an CSV file :Example: >>> from Backup import Backup >>> from ListUsers import ListUsers >>> L = ListUsers.ListUsers("/path/to/sqlite").get_fetch() >>> backup_db = Backup("stdout") >>> backup_db.backup_to_file(L) CSV-Separated list with headers in first row """ filename: str quoting: int dialect: str field_names: tuple def __init__(self, output: str, quoting: int = csv.QUOTE_NONNUMERIC, dialect: str = "excel"): """ Constructs the Backup object :param output: File name to backup to(set to stdout for stdout) :type output: str :param quoting: Set quoting for CSV Module :type quoting: int :param dialect: Set the CSV-Dialect. Defaults to excel, which is the classic CSV :type dialect: str """ self.setFilename(output) self.setQuoting(quoting) self.setDialect(dialect) self.setFieldnames(tuple(['id', 'username', 'email', 'name', 'pubkey', 'timestamp', 'status'])) def setDialect(self, dialect: str) -> None: """ Set dialect for Object :param dialect: Dialect to set for Object :type dialect: str :return: None :rtype: None """ self.dialect = dialect def setQuoting(self, quoting: int) -> None: """ Set quoting in the CSV(must be supported by the CSV Module!) :param quoting: Quoting Integer given by csv.QUOTE_* constants :type quoting: int :return: None :rtype: None """ self.quoting = quoting def setFilename(self, filename: str) -> None: """ Sets Filename to output to :param filename: Filename to output to(set stdout for stdout) :type filename: str :return: None :rtype: None """ self.filename = filename def setFieldnames(self, f_names: tuple) -> None: """ Set fieldname to process :param f_names: Fieldnames-Tuple :type f_names: tuple :return: None :rtype: None """ self.field_names = f_names def backup_to_file(self, fetched: list) -> bool: """Backup Userlist to File(or stdout) :param fetched: List of values to write out CSV-formatted :return: True, if success, None when not. :rtype: bool """ returner = io.StringIO() write_csv = csv.DictWriter(returner, fieldnames=self.field_names, quoting=self.quoting, dialect=self.dialect) write_csv.writeheader() write_csv.writerows(fetched) if self.filename == "stdout": print(returner.getvalue()) return True else: with open(self.filename, "w") as f: print(returner.getvalue(), file=f) return True if __name__ == "__main__": default_cmd.argparser.description += " - Backups Tilde Users to stdout or a file." args = default_cmd.argparser.parse_args() config = configparser.ConfigParser() config.read(args.config) try: L = ListUsers.ListUsers(config['DEFAULT']['applications_db'], unapproved=args.unapproved, approved=args.approved) fetch = L.get_fetch() B = Backup(args.file) B.backup_to_file(fetch) exit(0) except KeyboardInterrupt as e: pass