/sqlautocode/tests/base.py

https://code.google.com/p/sqlautocode/ · Python · 91 lines · 65 code · 20 blank · 6 comment · 2 complexity · 1f03f8f8c60d0be5761a9eebe4ddcd9d MD5 · raw file

  1. import os
  2. from sqlalchemy import *
  3. metadata = MetaData()
  4. environment = Table('environment', metadata,
  5. Column(u'environment_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=True, nullable=False),
  6. Column(u'environment_name', String(length=100, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  7. Column(u'database_host', String(length=100, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  8. Column(u'database_port', String(length=5, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  9. Column(u'database_sid', String(length=32, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  10. Column(u'database_user', String(length=100, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  11. Column(u'database_pass', String(length=100, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  12. )
  13. report = Table('report', metadata,
  14. Column(u'report_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=True, nullable=False),
  15. Column(u'environment_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  16. Column(u'report_name', String(length=50, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  17. Column(u'report_description', String(length=4000, convert_unicode=False, assert_unicode=None), primary_key=False),
  18. Column(u'deleted', Numeric(precision=1, scale=0, asdecimal=True), primary_key=False, nullable=False),
  19. Column(u'created_date', DateTime(timezone=False), primary_key=False, nullable=False),
  20. Column(u'created_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  21. Column(u'updated_date', DateTime(timezone=False), primary_key=False, nullable=False),
  22. Column(u'updated_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  23. Column(u'deleted_date', DateTime(timezone=False), primary_key=False),
  24. Column(u'deleted_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False),
  25. ForeignKeyConstraint([u'environment_id'], [u'environment.environment_id'], name='REPORT_FK_ENV_ID'),
  26. )
  27. ui_report = Table('ui_report', metadata,
  28. Column(u'ui_report_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=True, nullable=False),
  29. Column(u'report_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  30. Column(u'environment_id', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  31. Column(u'ui_report_name', String(length=100, convert_unicode=False, assert_unicode=None), primary_key=False, nullable=False),
  32. Column(u'ui_report_description', String(length=4000, convert_unicode=False, assert_unicode=None), primary_key=False),
  33. Column(u'enabled', Numeric(precision=1, scale=0, asdecimal=True), primary_key=False, nullable=False),
  34. Column(u'deleted', Numeric(precision=1, scale=0, asdecimal=True), primary_key=False, nullable=False),
  35. Column(u'created_date', DateTime(timezone=False), primary_key=False, nullable=False),
  36. Column(u'created_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  37. Column(u'updated_date', DateTime(timezone=False), primary_key=False, nullable=False),
  38. Column(u'updated_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False, nullable=False),
  39. Column(u'deleted_date', DateTime(timezone=False), primary_key=False),
  40. Column(u'deleted_by', Numeric(precision=10, scale=0, asdecimal=True), primary_key=False),
  41. ForeignKeyConstraint([u'report_id'], [u'report.report_id'], name='UI_REPORT_FK_REPORT_ID'),
  42. ForeignKeyConstraint([u'environment_id'], [u'environment.environment_id'], name='UI_REPORT_FK_ENV_ID'),
  43. )
  44. bound = False
  45. def make_test_db():
  46. global bound, metadata
  47. if not bound:
  48. testdb_filename = os.path.abspath(os.path.dirname(__file__))+'/data/testdb.db'
  49. #try:
  50. # os.remove(testdb_filename)
  51. #except OSError:
  52. # pass
  53. db = 'sqlite:///'+testdb_filename
  54. test_engine = create_engine(db)
  55. metadata.bind =test_engine
  56. #metadata.create_all()
  57. bound = True
  58. return metadata
  59. bound_multi = False
  60. metadata_multi = MetaData()
  61. def make_test_db_multi():
  62. global bound_multi, metadata_multi
  63. if not bound_multi:
  64. testdb_filename = os.path.abspath(os.path.dirname(__file__))+'/data/multi.db'
  65. #testdb_filename = os.path.abspath(os.path.dirname(__file__))+'/data/devdata.db'
  66. db = 'sqlite:///'+testdb_filename
  67. test_engine = create_engine(db)
  68. metadata_multi.bind =test_engine
  69. metadata_multi.reflect()
  70. bound_multi = True
  71. return metadata_multi