Added more utils, added model for workflow

taskflow/db/api.py

@@ -17,6 +17,8 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+"""Implementation of SQLAlchemy Backend"""
+
 from oslo.config import cfg
 
 from taskflow.common import config

taskflow/db/sqlalchemy/api.py

@@ -16,8 +16,15 @@
 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 #    License for the specific language governing permissions and limitations
 #    under the License.
+
 """Implementation of SQLAlchemy backend."""
 
 import logging
 
 LOG = logging.getLogger(__name__)
+
+def model_query(context, *args, **kwargs):
+    session = kwargs.get('session') or get_session()
+    query = session.query(*args)
+
+    return query

taskflow/db/sqlalchemy/session.py

@@ -84,6 +84,12 @@ def get_engine():
     _ENGINE = sqlalchemy.create_engine(_get_sql_connection(),
                                        **engine_args)
 
+    if 'mysql' in connection_dict.drivername:
+        sqlalchemy.event.listen(_ENGINE, 'checkout', ping_listener)
+    if 'sqlite' in connection_dict.drivername:
+        sqlalchemy.event.listen(_ENGINE, 'connect',
+                                synchronous_switch_listener)
+
     #TODO: Check to make sure engine connected
 
     return _ENGINE

taskflow/openstack/common/timeutils.py

@@ -0,0 +1,33 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright 2011 OpenStack LLC.
+# All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+"""
+Time related utilities and helper functions.
+"""
+
+import datetime
+
+def utcnow():
+    """Overridable version of utils.utcnow."""
+    if utcnow.override_time:
+        try:
+            return utcnow.override_time.pop(0)
+        except AttributeError:
+            return utcnow.override_time
+    return datetime.datetime.utcnow()
+
+utcnow.override_time = None

taskflow/openstack/common/uuidutils.py

@@ -0,0 +1,27 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright (c) 2012 Intel Corporation.
+# All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+"""
+UUID related utilities and helper functions.
+"""
+
+import uuid
+
+
+def generate_uuid():
+    return str(uuid.uuid4())
+