Initial commit of DB schema.
Initial commit of the event schema for the database. This includes models and alembic migration.
This commit is contained in:
commit
aa8fb55e87
30
.gitignore
vendored
Normal file
30
.gitignore
vendored
Normal file
@ -0,0 +1,30 @@
|
||||
*.py[cod]
|
||||
|
||||
# C extensions
|
||||
*.so
|
||||
|
||||
# Packages
|
||||
*.egg
|
||||
*.egg-info
|
||||
dist
|
||||
build
|
||||
eggs
|
||||
parts
|
||||
var
|
||||
sdist
|
||||
develop-eggs
|
||||
.installed.cfg
|
||||
lib
|
||||
lib64
|
||||
__pycache__
|
||||
|
||||
# Installer logs
|
||||
pip-log.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
.coverage
|
||||
.tox
|
||||
nosetests.xml
|
||||
|
||||
# Translations
|
||||
*.mo
|
202
LICENSE
Normal file
202
LICENSE
Normal file
@ -0,0 +1,202 @@
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "{}"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright {yyyy} {name of copyright owner}
|
||||
|
||||
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.
|
||||
|
2
MANIFEST.in
Normal file
2
MANIFEST.in
Normal file
@ -0,0 +1,2 @@
|
||||
include README.md
|
||||
include requirements.txt
|
5
README.md
Normal file
5
README.md
Normal file
@ -0,0 +1,5 @@
|
||||
winchester
|
||||
==========
|
||||
|
||||
An OpenStack notification event processing library based on persistant streams.
|
||||
|
50
alembic.ini
Normal file
50
alembic.ini
Normal file
@ -0,0 +1,50 @@
|
||||
# A generic, single database configuration.
|
||||
|
||||
[alembic]
|
||||
# path to migration scripts
|
||||
script_location = alembic
|
||||
|
||||
# template used to generate migration files
|
||||
# file_template = %%(rev)s_%%(slug)s
|
||||
|
||||
# set to 'true' to run the environment during
|
||||
# the 'revision' command, regardless of autogenerate
|
||||
# revision_environment = false
|
||||
|
||||
sqlalchemy.url = driver://user:pass@localhost/dbname
|
||||
|
||||
|
||||
# Logging configuration
|
||||
[loggers]
|
||||
keys = root,sqlalchemy,alembic
|
||||
|
||||
[handlers]
|
||||
keys = console
|
||||
|
||||
[formatters]
|
||||
keys = generic
|
||||
|
||||
[logger_root]
|
||||
level = WARN
|
||||
handlers = console
|
||||
qualname =
|
||||
|
||||
[logger_sqlalchemy]
|
||||
level = WARN
|
||||
handlers =
|
||||
qualname = sqlalchemy.engine
|
||||
|
||||
[logger_alembic]
|
||||
level = INFO
|
||||
handlers =
|
||||
qualname = alembic
|
||||
|
||||
[handler_console]
|
||||
class = StreamHandler
|
||||
args = (sys.stderr,)
|
||||
level = NOTSET
|
||||
formatter = generic
|
||||
|
||||
[formatter_generic]
|
||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
||||
datefmt = %H:%M:%S
|
72
alembic/env.py
Normal file
72
alembic/env.py
Normal file
@ -0,0 +1,72 @@
|
||||
from __future__ import with_statement
|
||||
from alembic import context
|
||||
from sqlalchemy import engine_from_config, pool
|
||||
from logging.config import fileConfig
|
||||
|
||||
# this is the Alembic Config object, which provides
|
||||
# access to the values within the .ini file in use.
|
||||
config = context.config
|
||||
|
||||
# Interpret the config file for Python logging.
|
||||
# This line sets up loggers basically.
|
||||
fileConfig(config.config_file_name)
|
||||
|
||||
# add your model's MetaData object here
|
||||
# for 'autogenerate' support
|
||||
# from myapp import mymodel
|
||||
# target_metadata = mymodel.Base.metadata
|
||||
from winchester.models import Base
|
||||
target_metadata = Base.metadata
|
||||
|
||||
# other values from the config, defined by the needs of env.py,
|
||||
# can be acquired:
|
||||
# my_important_option = config.get_main_option("my_important_option")
|
||||
# ... etc.
|
||||
|
||||
def run_migrations_offline():
|
||||
"""Run migrations in 'offline' mode.
|
||||
|
||||
This configures the context with just a URL
|
||||
and not an Engine, though an Engine is acceptable
|
||||
here as well. By skipping the Engine creation
|
||||
we don't even need a DBAPI to be available.
|
||||
|
||||
Calls to context.execute() here emit the given string to the
|
||||
script output.
|
||||
|
||||
"""
|
||||
url = config.get_main_option("sqlalchemy.url")
|
||||
context.configure(url=url)
|
||||
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
def run_migrations_online():
|
||||
"""Run migrations in 'online' mode.
|
||||
|
||||
In this scenario we need to create an Engine
|
||||
and associate a connection with the context.
|
||||
|
||||
"""
|
||||
engine = engine_from_config(
|
||||
config.get_section(config.config_ini_section),
|
||||
prefix='sqlalchemy.',
|
||||
poolclass=pool.NullPool)
|
||||
|
||||
connection = engine.connect()
|
||||
context.configure(
|
||||
connection=connection,
|
||||
target_metadata=target_metadata
|
||||
)
|
||||
|
||||
try:
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
finally:
|
||||
connection.close()
|
||||
|
||||
if context.is_offline_mode():
|
||||
run_migrations_offline()
|
||||
else:
|
||||
run_migrations_online()
|
||||
|
22
alembic/script.py.mako
Normal file
22
alembic/script.py.mako
Normal file
@ -0,0 +1,22 @@
|
||||
"""${message}
|
||||
|
||||
Revision ID: ${up_revision}
|
||||
Revises: ${down_revision}
|
||||
Create Date: ${create_date}
|
||||
|
||||
"""
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = ${repr(up_revision)}
|
||||
down_revision = ${repr(down_revision)}
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
${imports if imports else ""}
|
||||
|
||||
def upgrade():
|
||||
${upgrades if upgrades else "pass"}
|
||||
|
||||
|
||||
def downgrade():
|
||||
${downgrades if downgrades else "pass"}
|
53
alembic/versions/3ab6d7bf80cd_.py
Normal file
53
alembic/versions/3ab6d7bf80cd_.py
Normal file
@ -0,0 +1,53 @@
|
||||
"""Initial Event schema.
|
||||
|
||||
Revision ID: 3ab6d7bf80cd
|
||||
Revises: None
|
||||
Create Date: 2014-06-26 01:36:42.792353
|
||||
|
||||
"""
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = '3ab6d7bf80cd'
|
||||
down_revision = None
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from winchester import models
|
||||
|
||||
def upgrade():
|
||||
### commands auto generated by Alembic - please adjust! ###
|
||||
op.create_table('event_type',
|
||||
sa.Column('id', sa.Integer(), nullable=False),
|
||||
sa.Column('desc', sa.String(length=255), nullable=True),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
sa.UniqueConstraint('desc')
|
||||
)
|
||||
op.create_table('event',
|
||||
sa.Column('id', sa.Integer(), nullable=False),
|
||||
sa.Column('message_id', sa.String(length=50), nullable=True),
|
||||
sa.Column('generated', models.PreciseTimestamp(), nullable=True),
|
||||
sa.Column('event_type_id', sa.Integer(), nullable=True),
|
||||
sa.ForeignKeyConstraint(['event_type_id'], ['event_type.id'], ),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
sa.UniqueConstraint('message_id')
|
||||
)
|
||||
op.create_table('trait',
|
||||
sa.Column('event_id', sa.Integer(), nullable=False),
|
||||
sa.Column('name', sa.String(length=100), nullable=False),
|
||||
sa.Column('type', sa.Integer(), nullable=True),
|
||||
sa.Column('t_string', sa.String(length=255), nullable=True),
|
||||
sa.Column('t_float', sa.Float(), nullable=True),
|
||||
sa.Column('t_int', sa.Integer(), nullable=True),
|
||||
sa.Column('t_datetime', models.PreciseTimestamp(), nullable=True),
|
||||
sa.ForeignKeyConstraint(['event_id'], ['event.id'], ),
|
||||
sa.PrimaryKeyConstraint('event_id', 'name')
|
||||
)
|
||||
### end Alembic commands ###
|
||||
|
||||
|
||||
def downgrade():
|
||||
### commands auto generated by Alembic - please adjust! ###
|
||||
op.drop_table('trait')
|
||||
op.drop_table('event')
|
||||
op.drop_table('event_type')
|
||||
### end Alembic commands ###
|
3
requirements.txt
Normal file
3
requirements.txt
Normal file
@ -0,0 +1,3 @@
|
||||
alembic>=0.4.1
|
||||
enum34>=1.0
|
||||
SQLAlchemy>=0.9.6
|
36
setup.py
Normal file
36
setup.py
Normal file
@ -0,0 +1,36 @@
|
||||
import os
|
||||
from pip.req import parse_requirements
|
||||
from setuptools import setup, find_packages
|
||||
|
||||
|
||||
def read(fname):
|
||||
return open(os.path.join(os.path.dirname(__file__), fname)).read()
|
||||
|
||||
|
||||
req_file = os.path.join(os.path.dirname(__file__), "requirements.txt")
|
||||
install_reqs = [str(r.req) for r in parse_requirements(req_file)]
|
||||
|
||||
|
||||
setup(
|
||||
name='winchester',
|
||||
version='0.10',
|
||||
author='Monsyne Dragon',
|
||||
author_email='mdragon@rackspace.com',
|
||||
description=("An OpenStack notification event processing library."),
|
||||
license='Apache License (2.0)',
|
||||
keywords='OpenStack notifications events processing triggers',
|
||||
packages=find_packages(exclude=['tests']),
|
||||
classifiers=[
|
||||
'Development Status :: 3 - Alpha',
|
||||
'License :: OSI Approved :: Apache Software License',
|
||||
'Operating System :: POSIX :: Linux',
|
||||
'Programming Language :: Python :: 2.6',
|
||||
'Programming Language :: Python :: 2.7',
|
||||
],
|
||||
url='https://github.com/StackTach/winchester',
|
||||
scripts=[],
|
||||
long_description=read('README.md'),
|
||||
install_requires=install_reqs,
|
||||
|
||||
zip_safe=False
|
||||
)
|
3
test-requirements.txt
Normal file
3
test-requirements.txt
Normal file
@ -0,0 +1,3 @@
|
||||
mock>=1.0
|
||||
nose
|
||||
unittest2
|
15
tox.ini
Normal file
15
tox.ini
Normal file
@ -0,0 +1,15 @@
|
||||
[tox]
|
||||
envlist = py26,py27
|
||||
|
||||
[testenv]
|
||||
deps =
|
||||
-r{toxinidir}/requirements.txt
|
||||
-r{toxinidir}/test-requirements.txt
|
||||
|
||||
setenv = VIRTUAL_ENV={envdir}
|
||||
|
||||
commands =
|
||||
nosetests tests
|
||||
|
||||
sitepackages = False
|
||||
|
0
winchester/__init__.py
Normal file
0
winchester/__init__.py
Normal file
70
winchester/db.py
Normal file
70
winchester/db.py
Normal file
@ -0,0 +1,70 @@
|
||||
import sqlalchemy
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from winchester import models
|
||||
|
||||
|
||||
ENGINES = dict()
|
||||
SESSIONMAKERS = dict()
|
||||
|
||||
|
||||
def sessioned(func):
|
||||
def with_session(self, *args, **kw):
|
||||
if 'session' in kw:
|
||||
return func(self, *args, **kw)
|
||||
else:
|
||||
try:
|
||||
session = self.get_session()
|
||||
kw['session'] = session
|
||||
retval = func(self, *args, **kw)
|
||||
session.commit()
|
||||
return retval
|
||||
except:
|
||||
session.rollback()
|
||||
raise
|
||||
finally:
|
||||
session.close()
|
||||
return with_session
|
||||
|
||||
|
||||
class DBInterface(object):
|
||||
|
||||
def __init__(self, config):
|
||||
self.config = config
|
||||
self.db_url = config['url']
|
||||
|
||||
@property
|
||||
def engine(self):
|
||||
global ENGINES
|
||||
if self.db_url not in ENGINES:
|
||||
engine = sqlalchemy.create_engine(self.db_url)
|
||||
ENGINES[self.db_url] = engine
|
||||
return ENGINES[self.db_url]
|
||||
|
||||
@property
|
||||
def sessionmaker(self):
|
||||
global SESSIONMAKERS
|
||||
if self.db_url not in SESSIONMAKERS:
|
||||
maker = sessionmaker(bind=self.engine)
|
||||
SESSIONMAKERS[self.db_url] = maker
|
||||
return SESSIONMAKERS[self.db_url]
|
||||
|
||||
def get_session(self):
|
||||
return self.sessionmaker(expire_on_commit=False)
|
||||
|
||||
@sessioned
|
||||
def get_event_type(self, description, session=None):
|
||||
t = session.query(models.EventType).filter(models.EventType.desc == description).first()
|
||||
if t is None:
|
||||
t = models.EventType(description)
|
||||
session.add(t)
|
||||
return t
|
||||
|
||||
@sessioned
|
||||
def create_event(self, message_id, event_type, generated, traits, session=None):
|
||||
event_type = self.get_event_type(event_type, session=session)
|
||||
e = models.Event(message_id, event_type, generated)
|
||||
for name in traits:
|
||||
e[name] = traits[name]
|
||||
session.add(e)
|
||||
return e
|
237
winchester/models.py
Normal file
237
winchester/models.py
Normal file
@ -0,0 +1,237 @@
|
||||
from datetime import datetime
|
||||
|
||||
from enum import IntEnum
|
||||
|
||||
from sqlalchemy import event
|
||||
from sqlalchemy import literal_column
|
||||
from sqlalchemy import Column, Table, ForeignKey, Index, UniqueConstraint
|
||||
from sqlalchemy import Float, Boolean, Text, DateTime, Integer, String
|
||||
from sqlalchemy import cast, null, case
|
||||
from sqlalchemy.orm.interfaces import PropComparator
|
||||
from sqlalchemy.ext.hybrid import hybrid_property
|
||||
from sqlalchemy.dialects.mysql import DECIMAL
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
from sqlalchemy.ext.associationproxy import association_proxy
|
||||
from sqlalchemy.orm import backref
|
||||
from sqlalchemy.orm import relationship
|
||||
from sqlalchemy.orm.collections import attribute_mapped_collection
|
||||
from sqlalchemy.types import TypeDecorator, DATETIME
|
||||
|
||||
|
||||
class Datatype(IntEnum):
|
||||
none = 0
|
||||
string = 1
|
||||
int = 2
|
||||
float = 3
|
||||
datetime = 4
|
||||
|
||||
|
||||
class PreciseTimestamp(TypeDecorator):
|
||||
"""Represents a timestamp precise to the microsecond."""
|
||||
|
||||
impl = DATETIME
|
||||
|
||||
def load_dialect_impl(self, dialect):
|
||||
if dialect.name == 'mysql':
|
||||
return dialect.type_descriptor(DECIMAL(precision=20,
|
||||
scale=6,
|
||||
asdecimal=True))
|
||||
return dialect.type_descriptor(DATETIME())
|
||||
|
||||
def process_bind_param(self, value, dialect):
|
||||
if value is None:
|
||||
return value
|
||||
elif dialect.name == 'mysql':
|
||||
return utils.dt_to_decimal(value)
|
||||
return value
|
||||
|
||||
def process_result_value(self, value, dialect):
|
||||
if value is None:
|
||||
return value
|
||||
elif dialect.name == 'mysql':
|
||||
return utils.decimal_to_dt(value)
|
||||
return value
|
||||
|
||||
|
||||
class ProxiedDictMixin(object):
|
||||
"""Adds obj[name] access to a mapped class.
|
||||
|
||||
This class basically proxies dictionary access to an attribute
|
||||
called ``_proxied``. The class which inherits this class
|
||||
should have an attribute called ``_proxied`` which points to a dictionary.
|
||||
|
||||
"""
|
||||
|
||||
def __len__(self):
|
||||
return len(self._proxied)
|
||||
|
||||
def __iter__(self):
|
||||
return iter(self._proxied)
|
||||
|
||||
def __getitem__(self, name):
|
||||
return self._proxied[name]
|
||||
|
||||
def __contains__(self, name):
|
||||
return name in self._proxied
|
||||
|
||||
def __setitem__(self, name, value):
|
||||
self._proxied[name] = value
|
||||
|
||||
def __delitem__(self, name):
|
||||
del self._proxied[name]
|
||||
|
||||
|
||||
class PolymorphicVerticalProperty(object):
|
||||
"""A name/value pair with polymorphic value storage."""
|
||||
|
||||
def __init__(self, name, value=None):
|
||||
self.name = name
|
||||
self.value = value
|
||||
|
||||
@hybrid_property
|
||||
def value(self):
|
||||
fieldname, discriminator = self.type_map[self.type]
|
||||
if fieldname is None:
|
||||
return None
|
||||
else:
|
||||
return getattr(self, fieldname)
|
||||
|
||||
@value.setter
|
||||
def value(self, value):
|
||||
py_type = type(value)
|
||||
fieldname, discriminator = self.type_map[py_type]
|
||||
|
||||
self.type = discriminator
|
||||
if fieldname is not None:
|
||||
setattr(self, fieldname, value)
|
||||
|
||||
@value.deleter
|
||||
def value(self):
|
||||
self._set_value(None)
|
||||
|
||||
@value.comparator
|
||||
class value(PropComparator):
|
||||
"""A comparator for .value, builds a polymorphic comparison via CASE.
|
||||
|
||||
"""
|
||||
def __init__(self, cls):
|
||||
self.cls = cls
|
||||
|
||||
def _case(self):
|
||||
pairs = set(self.cls.type_map.values())
|
||||
whens = [
|
||||
(
|
||||
literal_column("'%s'" % discriminator),
|
||||
cast(getattr(self.cls, attribute), String)
|
||||
) for attribute, discriminator in pairs
|
||||
if attribute is not None
|
||||
]
|
||||
return case(whens, self.cls.type, null())
|
||||
def __eq__(self, other):
|
||||
return self._case() == cast(other, String)
|
||||
def __ne__(self, other):
|
||||
return self._case() != cast(other, String)
|
||||
|
||||
def __repr__(self):
|
||||
return '<%s %r=%r>' % (self.__class__.__name__, self.name, self.value)
|
||||
|
||||
|
||||
@event.listens_for(PolymorphicVerticalProperty, "mapper_configured", propagate=True)
|
||||
def on_new_class(mapper, cls_):
|
||||
"""Add type lookup info for polymorphic value columns.
|
||||
"""
|
||||
|
||||
info_dict = {}
|
||||
info_dict[type(None)] = (None, Datatype.none)
|
||||
info_dict[Datatype.none] = (None, Datatype.none)
|
||||
|
||||
for k in mapper.c.keys():
|
||||
col = mapper.c[k]
|
||||
if 'type' in col.info:
|
||||
python_type, discriminator = col.info['type']
|
||||
info_dict[python_type] = (k, discriminator)
|
||||
info_dict[discriminator] = (k, discriminator)
|
||||
cls_.type_map = info_dict
|
||||
|
||||
|
||||
Base = declarative_base()
|
||||
|
||||
|
||||
class Trait(PolymorphicVerticalProperty, Base):
|
||||
__tablename__ = 'trait'
|
||||
__table_args__ = (
|
||||
Index('ix_trait_t_int', 't_int'),
|
||||
Index('ix_trait_t_string', 't_string'),
|
||||
Index('ix_trait_t_datetime', 't_datetime'),
|
||||
Index('ix_trait_t_float', 't_float'),
|
||||
)
|
||||
event_id = Column(Integer, ForeignKey('event.id'), primary_key=True)
|
||||
name = Column(String(100), primary_key=True)
|
||||
type = Column(Integer)
|
||||
|
||||
|
||||
t_string = Column(String(255), info=dict(type=(str, Datatype.string)),
|
||||
nullable=True, default=None)
|
||||
t_float = Column(Float, info=dict(type=(float, Datatype.float)),
|
||||
nullable=True, default=None)
|
||||
t_int = Column(Integer, info=dict(type=(int, Datatype.int)),
|
||||
nullable=True, default=None)
|
||||
t_datetime = Column(PreciseTimestamp(), info=dict(type=(datetime, Datatype.datetime)),
|
||||
nullable=True, default=None)
|
||||
|
||||
def __repr__(self):
|
||||
return "<Trait(%s) %s=%s/%s/%s/%s on %s>" % (self.name,
|
||||
self.type,
|
||||
self.t_string,
|
||||
self.t_float,
|
||||
self.t_int,
|
||||
self.t_datetime,
|
||||
self.event_id)
|
||||
|
||||
|
||||
class EventType(Base):
|
||||
"""Types of event records."""
|
||||
__tablename__ = 'event_type'
|
||||
|
||||
id = Column(Integer, primary_key=True)
|
||||
desc = Column(String(255), unique=True)
|
||||
|
||||
def __init__(self, event_type):
|
||||
self.desc = event_type
|
||||
|
||||
def __repr__(self):
|
||||
return "<EventType: %s>" % self.desc
|
||||
|
||||
|
||||
class Event(ProxiedDictMixin, Base):
|
||||
__tablename__ = 'event'
|
||||
__table_args__ = (
|
||||
Index('ix_event_message_id', 'message_id'),
|
||||
Index('ix_event_type_id', 'event_type_id'),
|
||||
Index('ix_event_generated', 'generated')
|
||||
)
|
||||
id = Column(Integer, primary_key=True)
|
||||
message_id = Column(String(50), unique=True)
|
||||
generated = Column(PreciseTimestamp())
|
||||
|
||||
event_type_id = Column(Integer, ForeignKey('event_type.id'))
|
||||
event_type = relationship("EventType", backref=backref('event_type'))
|
||||
|
||||
traits = relationship("Trait",
|
||||
collection_class=attribute_mapped_collection('name'))
|
||||
_proxied = association_proxy("traits", "value",
|
||||
creator=lambda name, value: Trait(name=name, value=value))
|
||||
|
||||
def __init__(self, message_id, event_type, generated):
|
||||
|
||||
self.message_id = message_id
|
||||
self.event_type = event_type
|
||||
self.generated = generated
|
||||
|
||||
def __repr__(self):
|
||||
return "<Event %s ('Event: %s %s, Generated: %s')>" % (self.id,
|
||||
self.message_id,
|
||||
self.event_type,
|
||||
self.generated)
|
||||
|
||||
|
8
winchester/trigger_manager.py
Normal file
8
winchester/trigger_manager.py
Normal file
@ -0,0 +1,8 @@
|
||||
from winchester.db import DBInterface
|
||||
|
||||
|
||||
class TriggerManager(object):
|
||||
|
||||
def __init__(self, config):
|
||||
self.config = config
|
||||
self.db = DBInterface(config['database'])
|
Loading…
Reference in New Issue
Block a user