コード例 #1
0
ファイル: setting.py プロジェクト: JervyShi/verify_zip_file
class Setting(object):
    def __init__(self, cfg_path):
        super(Setting, self).__init__()
        self.cf = MyConfigParser(cfg_path)

    def get_default_config(self):
        return self.cf["basic"]["default_config"]

    def get_work_path(self):
        return self.cf["basic"]["work_path"]

    def get_file_config_path(self, file_name):
        for name in self.cf["file_config_mapper"]:
            value = self.cf["file_config_mapper"][name]
            if file_name.find(name) >= 0:
                return value
        return None

    def set_config_mapper(self, name, value):
        self.cf["file_config_mapper"][name] = value
        self.cf.write()

    def del_config_mapper(self, name):
        del self.cf["file_config_mapper"][name]
        self.cf.write()
コード例 #2
0
 def extract(self):
     members = self.zf.infolist()
     if self.show_info:
         self.start_extract()
     total = 0
     for zip_info in members:
         total += zip_info.file_size
         if not self.regex_util or not self.regex_util.do_match(spit_filename(zip_info.filename, True)):
             if len(zip_info.filename) + len(self.work_path) + 1 < 255:
                 self.zf.extract(zip_info.filename, self.work_path)
                 if self.eu_text and spit_filename(zip_info.filename, True) == "important.properties":
                     file_path = os.path.join(self.work_path, zip_info.filename)
                     try:
                         cf = MyConfigParser(file_path, file_error=True)
                         regrex = ["\$\{" + key + "\}" for key in cf.keys()]
                         self.eu_text.add_regex(regrex)
                     except Exception as exe:
                         print exe
                         print self.work_path + "/" + zip_info.filename
                 if self.show_info:
                     self.update_extract(total)
             else:
                 print "path len > 255   ", self.work_path, zip_info.filename
         else:
             pass
     if self.show_info:
         self.finish_extract()
     self.zf.close()
コード例 #3
0
ファイル: model.py プロジェクト: purepitch/trove
 def __init__(self):
     """
     Initializes the Model class for trove with empty member variables.
     """
     self.program_name = ""
     self.version = ""
     self.config = MyConfigParser()
     self.entries = MyConfigParser()
     self.start_marker = False
     self.end_marker = False
     self.entry_dict = {}
     self.trove_dir = os.path.join(os.getenv('HOME'), '.trove')
     self.config_file_name = 'trove.conf'
     return None
コード例 #4
0
from gettext import gettext

from sys import path as sys_path
sys_path.append("money/reusing")
from casts import list2string, string2list_of_strings, string2list_of_integers, str2bool
from datetime_functions import string2dtnaive
from text_inputs import input_string, input_YN
from myconfigparser import MyConfigParser

_=gettext


if __name__ == "__main__":

    print("Hidden settings are going to be generated in /etc/django_money/settings.conf")
    config=MyConfigParser("/etc/django_money/settings.conf")

    ans=input_YN("Do you want to change database settings?")
    if ans is True:
        ans = input_string("Add you database server", "127.0.0.1")
        config.set("db", "server", ans)
        ans = input_string("Add you database port", "5432")
        config.set("db", "port", ans)
        ans = input_string("Add your database name", "mylibrary")
        config.set("db", "db", ans)
        ans = input_string("Add you database user", "postgres")
        config.cset("db", "user", ans)
        ans = input_string("Add you database password", "your_pass")
        config.cset("db", "password", ans)

    ans=input_YN("Do you want to change smtp mail server settings?")
コード例 #5
0
ファイル: setting.py プロジェクト: JervyShi/verify_zip_file
 def __init__(self, cfg_path):
     super(Setting, self).__init__()
     self.cf = MyConfigParser(cfg_path)
コード例 #6
0
ファイル: model.py プロジェクト: purepitch/trove
class Model():
    """
    Model for the trove program.
    This class handles all business logic.
    """
    def __init__(self):
        """
        Initializes the Model class for trove with empty member variables.
        """
        self.program_name = ""
        self.version = ""
        self.config = MyConfigParser()
        self.entries = MyConfigParser()
        self.start_marker = False
        self.end_marker = False
        self.entry_dict = {}
        self.trove_dir = os.path.join(os.getenv('HOME'), '.trove')
        self.config_file_name = 'trove.conf'
        return None

    def new_entry(self):
        """
        Initialise and return a new ``TroveEntry``.
        """
        return TroveEntry()

    def calculate_hash(self, entry):
        """
        Calculate and return the entry's SHA1 hash.
        """
        return hashlib.sha1(entry.name + entry.user + entry.passwd +
                            entry.helptext + entry.description).hexdigest()

    def check_choice(self, choice_type, choice, maximum = 0):
        """
        Method to verify a user input. Two modes are available:
        1) Integer mode allows for verification of a given integer
        to (a) be an integer indeed and (b) lie within a given range.
        Returns True in this case, False in all other cases.
        2) Boolean mode returns True if the user types either 'y',
        'Y', 'yes' or 'Yes', and returns False in all other cases.
        """
        if choice_type == 'integer':
            if not self.is_int(choice):
                return False
            if (int(choice) > maximum):
                return False
            elif (int(choice) < 1):
                return False
            else:
                return True
        elif choice_type == 'boolean':
            if choice == 'y' or choice == 'Y':
                return True
            elif choice == 'yes' or choice == 'Yes':
                return True
            else:
                return  False
        else:
            return False

    def decrypt_file(self, filename, passwd):
        """
        Decrypt the encrypted secrets file.
        """
        f = open(os.devnull, 'w')
        decrypt = subprocess.Popen(['bcrypt', filename], stdin=subprocess.PIPE, stdout=f, stderr=f)
        decrypt.stdin.write(passwd + "\n")
        decrypt.wait()
        f.close()

    def is_int(self, s):
        """
        Takes a string s and checks if is an integer
        """
        try:
            int(s)
            return True
        except ValueError:
            return False

    def get_entries(self, encrypted_file, passphrase):
        """
        Decrypts encrypted_file with a given master passphrase, reads in all
        entries (by calling extract_entries method) and encrypts the password
        file again with the same master passphrase.
        """
        filesize = 0
        self.decrypt_file(encrypted_file, passphrase)
        decrypted_filename = encrypted_file.rstrip('.bfe')
        if os.path.isfile(decrypted_filename):
            filesize = os.path.getsize(decrypted_filename)
        if filesize > 0:
            self.extract_entries(decrypted_filename)
            self.encrypt_file(decrypted_filename, passphrase)
        else:
            # workaround for bcrypt.  If the master passphrase is incorrect,
            # bcrypt creates an *empty* file (0 bytes) (and doesn't warn
            # about this being the case), which would then be encrypted by
            # trove and the old database would be overwritten with this
            # empty file.  Deleting the zero size password file is a
            # workaround so that the database with real data in doesn't get
            # overwritten with empty data.
            os.system("rm -f " + decrypted_filename)

    def write_encrypted_file(self, encryptedfile, masterpasswd):
        """
        Write the encrypted data to file using the given master password.
        """
        passwdfile = encryptedfile.rstrip('.bfe')
        ef = open(passwdfile, 'w')
        ef.write("# Auto generated by " + self.program_name + " " +
                 self.version + "\n")
        now = datetime.datetime.now()
        ef.write("# Date: " + \
                datetime.date.strftime(now, "%Y-%m-%d %H:%M:%S") + "\n\n")
        ef.write("# ### TROVE START MARKER ###\n\n")
        for key in self.entry_dict:
            entry = self.entry_dict[key]
            ef.write("[" + entry.name + "]\n")
            ef.write("user: "******"\n")
            ef.write("password: "******"\n")
            ef.write("help: " + entry.helptext + "\n")
            ef.write("description: " + entry.description + "\n\n")
        ef.write("# ### TROVE END MARKER ###\n")
        ef.close()
        os.system("rm -f " + encryptedfile)
        self.encrypt_file(passwdfile, masterpasswd)

    def encrypt_file(self, filename, passwd):
        """
        Encrypt the secrets file.
        """
        f = open(os.devnull, 'w')
        encrypt = subprocess.Popen(['bcrypt', filename], stdin=subprocess.PIPE, stdout=f, stderr=f)
        encrypt.stdin.write(passwd + "\n" + passwd + "\n")
        encrypt.wait()
        f.close()

    def extract_entries(self, filename):
        """
        Reads in decrypted password file and fills self.entry_dict with
        TroveEntry objects. Keys of this dictionary are the SHA1 hashes
        of the entries calculated with self.calculate_hash().
        """
        fh = open(filename, 'r')
        fhlines = fh.readlines()
        fh.close()
        for line in fhlines:
            line = line.strip()
            if (re.search('TROVE START MARKER', line)):
                self.start_marker = True
            if (re.search('TROVE END MARKER', line)):
                self.end_marker = True
        if self.start_marker and self.end_marker:
            self.entries.read(filename)
            for entry in self.entries.sections():
                myentry = TroveEntry()
                myentry.name = entry
                if 'user' in self.entries.options(entry):
                    myentry.user = self.entries.get(entry, 'user')
                if 'password' in self.entries.options(entry):
                    myentry.passwd = self.entries.get(entry, 'password')
                if 'help' in self.entries.options(entry):
                    myentry.helptext = self.entries.get(entry, 'help')
                if 'description' in self.entries.options(entry):
                    myentry.description = self.entries.get(entry, 'description')
                myentry.eid = self.calculate_hash(myentry)
                self.entry_dict[myentry.eid] = myentry
        return

    def search(self, search_term):
        """
        Searches name fields in self.entry_dict for search_term.
        Both strings are converted to lower case.
        Returns a list of Trove entry objects, where the regular
        expression search has been successful.
        It returns a list and not a dictionary, because later the list is sorted
        by entry names and displayed. The user then picks a result by choosing
        a number in the menu that shows up. Sorting would be impossible with a
        dictionary.
        """
        result_list = []
        for key in self.entry_dict.keys():
            normalized_entry_name = (self.entry_dict[key].name).lower()
            if re.search(search_term.lower(), normalized_entry_name):
                result_list.append(self.entry_dict[key])
        return result_list

    def password_search(self, search_term):
        """
        Searches password fields in self.entry_dict for search_term.
        Returns a list of Trove entry objects, where the regular
        expression search has been successful.
        """
        result_list = []
        for key in self.entry_dict:
            if re.search(search_term, (self.entry_dict[key].passwd)):
                result_list.append(self.entry_dict[key])
        return result_list

    def execute(self, command):
        """
        Run the given command in a subprocess.
        """
        proc = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        proc.wait()
        return_value = proc.poll()
        output = proc.communicate()[0]
        return return_value, output

    def return_all(self):
        """
        Returns a result list with all entries of the trove.
        """
        result_list = []
        for key in self.entry_dict.keys():
            result_list.append(self.entry_dict[key])
        return result_list