From 60ab4c2c127689e13fe1f448781aa663fb672289 Mon Sep 17 00:00:00 2001 From: Oleg Broytman Date: Wed, 23 Dec 2015 00:07:18 +0300 Subject: [PATCH] Add insert_name Insert a unique name if it doesn't exist. --- m_librarian/db.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/m_librarian/db.py b/m_librarian/db.py index 199f967..7cd53a1 100755 --- a/m_librarian/db.py +++ b/m_librarian/db.py @@ -1,12 +1,12 @@ #! /usr/bin/env python __all__ = ['Author', 'Book', 'Extension', 'Genre', 'Language', - 'init_db', + 'init_db', 'insert_name', ] import os from sqlobject import SQLObject, StringCol, UnicodeCol, IntCol, BoolCol, \ - ForeignKey, DateCol, connectionForURI, sqlhub, dberrors + ForeignKey, DateCol, connectionForURI, sqlhub, SQLObjectNotFound, dberrors from .config import ml_conf try: @@ -91,6 +91,13 @@ def init_db(): return +def insert_name(table, name): + try: + return table.byName(name) + except SQLObjectNotFound: + return table(name=name, count=0) + + def test(): print "DB dirs:", db_dirs if db_uri: -- 2.39.5