gryf/pygtktalog
1
0
Fork 0
mirror of https://github.com/gryf/pygtktalog.git synced 2025-12-17 11:30:19 +01:00
Code Issues Projects Releases Wiki Activity

Add ability to change name of the root item in the catalog.

Browse Source
This commit is contained in:
gryf
2022-09-25 21:13:14 +02:00
parent 7281f9bbbb
commit 54c24b18b1
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
pycatalog/__init__.py
View File

@@ -164,7 +164,7 @@ class Iface(object):
# scanob.update_files(node.id) # scanob.update_files(node.id)
scanob.update_files(node.id, self.engine) scanob.update_files(node.id, self.engine)
def create(self, dir_to_add): def create(self, dir_to_add, label=None):
"""Create new database""" """Create new database"""
self.root = dbo.File() self.root = dbo.File()
self.root.id = 1 self.root.id = 1
@@ -174,19 +174,17 @@ class Iface(object):
self.root.type = 0 self.root.type = 0
self.root.parent_id = 1 self.root.parent_id = 1
if not self.dry_run: if not self.dry_run:
self.sess.add(self.root) self.sess.add(self.root)
self.sess.add(config)
self.sess.commit() self.sess.commit()
print(misc.colorize("Creating new db against directory `%s'" % print(misc.colorize("Creating new db against directory `%s'" %
dir_to_add, 'white')) dir_to_add, 'white'))
if not self.dry_run: if not self.dry_run:
scanob = scan.Scan(dir_to_add) scanob = scan.Scan(dir_to_add)
scanob.add_files(self.engine) scanob.add_files(label=label)
def add(self, dir_to_add): def add(self, dir_to_add, label=None):
"""Add new directory to the db""" """Add new directory to the db"""
self.root = self.sess.query(dbo.File) self.root = self.sess.query(dbo.File)
self.root = self.root.filter(dbo.File.type == 0).first() self.root = self.root.filter(dbo.File.type == 0).first()
@@ -197,7 +195,7 @@ class Iface(object):
print(misc.colorize("Adding directory `%s'" % dir_to_add, 'white')) print(misc.colorize("Adding directory `%s'" % dir_to_add, 'white'))
if not self.dry_run: if not self.dry_run:
scanob = scan.Scan(dir_to_add) scanob = scan.Scan(dir_to_add)
scanob.add_files() scanob.add_files(label=label)
def _annotate(self, item, search_words): def _annotate(self, item, search_words):
""" """
@@ -273,14 +271,14 @@ def update_db(args):
def add_dir(args): def add_dir(args):
"""Add""" """Add"""
obj = Iface(args.db, args.pretend, args.debug) obj = Iface(args.db, args.pretend, args.debug)
obj.add(args.dir_to_add) obj.add(args.dir_to_add, args.label)
obj.close() obj.close()
def create_db(args): def create_db(args):
"""List""" """List"""
obj = Iface(args.db, args.pretend, args.debug) obj = Iface(args.db, args.pretend, args.debug)
obj.create(args.dir_to_add, args.imagedir) obj.create(args.dir_to_add, args.label)
obj.close() obj.close()
@@ -322,6 +320,8 @@ def main():
create = subparser.add_parser('create') create = subparser.add_parser('create')
create.add_argument('db') create.add_argument('db')
create.add_argument('dir_to_add') create.add_argument('dir_to_add')
create.add_argument('-l', '--label', help='Add label as the root item of '
'the added directory')
create.add_argument('-p', '--pretend', help="Don't do the action, just " create.add_argument('-p', '--pretend', help="Don't do the action, just "
"give the info what would gonna to happen.", "give the info what would gonna to happen.",
action='store_true', default=False) action='store_true', default=False)
@@ -337,6 +337,8 @@ def main():
action='store_true', default=False) action='store_true', default=False)
add.add_argument('-d', '--debug', help='Turn on debug', add.add_argument('-d', '--debug', help='Turn on debug',
action='store_true', default=False) action='store_true', default=False)
add.add_argument('-l', '--label', help='Add label as the root item of the '
'added directory')
add.set_defaults(func=add_dir) add.set_defaults(func=add_dir)
find = subparser.add_parser('find') find = subparser.add_parser('find')

2
pycatalog/scan.py
View File

@@ -47,7 +47,7 @@ class Scan(object):
self.files_count = self._get_files_count() self.files_count = self._get_files_count()
self.current_count = 0 self.current_count = 0
def add_files(self, engine=None): def add_files(self, label=None):
""" """
Returns list, which contain object, modification date and file Returns list, which contain object, modification date and file
size. size.