Module a8654abcec65_unify_scm_source_types
[hide private]
[frames] | no frames]

Source Code for Module a8654abcec65_unify_scm_source_types

 1  """unify-scm-source-types 
 2   
 3  Revision ID: a8654abcec65 
 4  Revises: fdec9947f8a1 
 5  Create Date: 2017-10-13 14:49:22.183416 
 6   
 7  """ 
 8   
 9  # revision identifiers, used by Alembic. 
10  revision = 'a8654abcec65' 
11  down_revision = 'fdec9947f8a1' 
12   
13  from alembic import op 
14  import sqlalchemy as sa 
15  import json 
16   
17  from coprs import models, db 
18   
19 -def upgrade():
20 session = sa.orm.sessionmaker(bind=op.get_bind())() 21 22 tito_build_rows = session.execute( 23 "SELECT * FROM build WHERE source_type=:param", 24 {"param": 3}) 25 26 for build in tito_build_rows: 27 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 28 new_source_dict = { 29 'type': 'git', 30 'clone_url': source_dict.get('git_url') or '', 31 'committish': source_dict.get('git_branch') or '', 32 'subdirectory': source_dict.get('git_dir') or '', 33 'spec': '', 34 'srpm_build_method': 'tito_test' if source_dict.get('tito_test') else 'tito', 35 } 36 new_source_json = json.dumps(new_source_dict) 37 new_source_type = 8 38 session.execute( 39 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 40 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']}) 41 42 mock_build_rows = session.execute( 43 "SELECT * FROM build WHERE source_type=:param", 44 {"param": 4}) 45 46 for build in mock_build_rows: 47 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 48 new_source_dict = { 49 'type': source_dict.get('scm_type') or 'git', 50 'clone_url': source_dict.get('scm_url') or '', 51 'committish': source_dict.get('scm_branch') or '', 52 'subdirectory': source_dict.get('scm_subdir') or '', 53 'spec': source_dict.get('spec') or '', 54 'srpm_build_method': 'rpkg', 55 } 56 new_source_json = json.dumps(new_source_dict) 57 new_source_type = 8 58 session.execute( 59 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 60 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']}) 61 62 fedpkg_build_rows = session.execute( 63 "SELECT * FROM build WHERE source_type=:param", 64 {"param": 7}) 65 66 for build in fedpkg_build_rows: 67 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 68 new_source_dict = { 69 'type': 'git', 70 'clone_url': source_dict.get('clone_url') or '', 71 'committish': source_dict.get('branch') or '', 72 'subdirectory': '', 73 'spec': '', 74 'srpm_build_method': 'rpkg', 75 } 76 new_source_json = json.dumps(new_source_dict) 77 new_source_type = 8 78 session.execute( 79 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 80 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']})
81 82
83 -def downgrade():
84 pass
85