Home Explore Help
Sign In
221V
/
flaskbb
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: 442380f668
Branches Tags
flaskbb
/
migrations
/
d9530a529b3f_add_timezone_awareness_for_datetime.py

d9530a529b3f_add_timezone_awareness_for_datetime.py 6.7 KB
History Raw

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 
  1. """add timezone awareness for datetime objects
    2. 

  2. 

  3. Revision ID: d9530a529b3f
    4. 

  4. Revises: 221d918aa9f0
    5. 

  5. Create Date: 2016-06-21 09:39:38.348519
    6. 

  6. 

  7. """
    8. 

  8. from alembic import op
    9. 

  9. import sqlalchemy as sa
    10. 

  10. import flaskbb
    11. 

  11. 

  12. # revision identifiers, used by Alembic.
    13. 

  13. revision = 'd9530a529b3f'
    14. 

  14. down_revision = '221d918aa9f0'
    15. 

  15. 

  16. 

  17. def upgrade():
    18. 

  18.     # batch_alter_table will only use the 'recreate' style on SQLite!
    19. 

  19. 

  20.     # user/models.py
    21. 

  21.     with op.batch_alter_table('users') as batch_op:
    22. 

  22.         batch_op.alter_column('date_joined', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    23. 

  23.         batch_op.alter_column('lastseen', existing_type=sa.DateTime(), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    24. 

  24.         batch_op.alter_column('birthday', existing_type=sa.DateTime(), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    25. 

  25.         batch_op.alter_column('last_failed_login', existing_type=sa.DateTime(), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    26. 

  26. 

  27.     # message/models.py
    28. 

  28.     with op.batch_alter_table('conversations') as batch_op:
    29. 

  29.         batch_op.alter_column('date_created', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    30. 

  30.     with op.batch_alter_table('messages') as batch_op:
    31. 

  31.         batch_op.alter_column('date_created', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    32. 

  32. 

  33.     # forum/models.py
    34. 

  34.     with op.batch_alter_table('topicsread') as batch_op:
    35. 

  35.         batch_op.alter_column('last_read', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    36. 

  36.     with op.batch_alter_table('forumsread') as batch_op:
    37. 

  37.         batch_op.alter_column('last_read', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    38. 

  38.         batch_op.alter_column('cleared', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    39. 

  39.     with op.batch_alter_table('reports') as batch_op:
    40. 

  40.         batch_op.alter_column('reported', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    41. 

  41.         batch_op.alter_column('zapped', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    42. 

  42.     with op.batch_alter_table('posts') as batch_op:
    43. 

  43.         batch_op.alter_column('date_created', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    44. 

  44.         batch_op.alter_column('date_modified', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    45. 

  45.     with op.batch_alter_table('topics') as batch_op:
    46. 

  46.         batch_op.alter_column('date_created', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    47. 

  47.         batch_op.alter_column('last_updated', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    48. 

  48.     with op.batch_alter_table('forums') as batch_op:
    49. 

  49.         batch_op.alter_column('last_post_created', existing_type=sa.DateTime(timezone=False), type_=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    50. 

  50. 

  51. 

  52. def downgrade():
    53. 

  53. 

  54.         # user/models.py
    55. 

  55.     with op.batch_alter_table('users') as batch_op:
    56. 

  56.         batch_op.alter_column('date_joined', type_=sa.DateTime(), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    57. 

  57.         batch_op.alter_column('lastseen', type_=sa.DateTime(), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    58. 

  58.         batch_op.alter_column('birthday', type_=sa.DateTime(), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    59. 

  59.         batch_op.alter_column('last_failed_login', type_=sa.DateTime(), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    60. 

  60. 

  61.         # message/models.py
    62. 

  62.     with op.batch_alter_table('conversations') as batch_op:
    63. 

  63.         batch_op.alter_column('date_created', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    64. 

  64.     with op.batch_alter_table('messages') as batch_op:
    65. 

  65.         batch_op.alter_column('date_created', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    66. 

  66. 

  67.         # forum/models.py
    68. 

  68.     with op.batch_alter_table('topicsread') as batch_op:
    69. 

  69.         batch_op.alter_column('last_read', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    70. 

  70.     with op.batch_alter_table('forumsread') as batch_op:
    71. 

  71.         batch_op.alter_column('last_read', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    72. 

  72.         batch_op.alter_column('cleared', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    73. 

  73.     with op.batch_alter_table('reports') as batch_op:
    74. 

  74.         batch_op.alter_column('reported', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    75. 

  75.         batch_op.alter_column('zapped', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    76. 

  76.     with op.batch_alter_table('posts') as batch_op:
    77. 

  77.         batch_op.alter_column('date_created', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    78. 

  78.         batch_op.alter_column('date_modified', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    79. 

  79.     with op.batch_alter_table('topics') as batch_op:
    80. 

  80.         batch_op.alter_column('date_created', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    81. 

  81.         batch_op.alter_column('last_updated', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    82. 

  82.     with op.batch_alter_table('forums') as batch_op:
    83. 

  83.         batch_op.alter_column('last_post_created', type_=sa.DateTime(timezone=False), existing_type=flaskbb.utils.database.UTCDateTime(timezone=True), existing_nullable=True)
    84.