Commit 9fa37e4b authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

added FileRecorder.Record

parent c2468010
Loading
Loading
Loading
Loading
+6 −3
Original line number Diff line number Diff line
@@ -2,7 +2,6 @@ from collections import namedtuple
from astropy import table
import os


File = namedtuple(
    "FileRecord",
    ["filepath", "db", "comment", "existence"]
@@ -52,8 +51,12 @@ class FileRecorder(list):
    test002.txt True Test file 2     False
    """

    def __init__(self):
        super(FileRecorder, self).__init__()
    def __init__(self, *args, **kwargs):
        super(FileRecorder, self).__init__(*args, **kwargs)

    @staticmethod
    def Record(filepath: str = "", db: bool = False, comment: str = ""):
        return File(filepath=filepath, db=db, comment=comment, existence=os.path.exists(filepath))

    def add_record(self, filepath: str = "", db: bool = False, comment: str = ""):
        existence = os.path.exists(filepath)