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   
18 -def upgrade():
19 session = sa.orm.sessionmaker(bind=op.get_bind())() 20 21 tito_build_rows = session.execute( 22 "SELECT * FROM build WHERE source_type=:param", 23 {"param": 3}) 24 25 for build in tito_build_rows: 26 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 27 new_source_dict = { 28 'type': 'git', 29 'clone_url': source_dict.get('git_url') or '', 30 'committish': source_dict.get('git_branch') or '', 31 'subdirectory': source_dict.get('git_dir') or '', 32 'spec': '', 33 'srpm_build_method': 'tito_test' if source_dict.get('tito_test') else 'tito', 34 } 35 new_source_json = json.dumps(new_source_dict) 36 new_source_type = 8 37 session.execute( 38 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 39 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']}) 40 41 mock_build_rows = session.execute( 42 "SELECT * FROM build WHERE source_type=:param", 43 {"param": 4}) 44 45 for build in mock_build_rows: 46 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 47 new_source_dict = { 48 'type': source_dict.get('scm_type') or 'git', 49 'clone_url': source_dict.get('scm_url') or '', 50 'committish': source_dict.get('scm_branch') or '', 51 'subdirectory': source_dict.get('scm_subdir') or '', 52 'spec': source_dict.get('spec') or '', 53 'srpm_build_method': 'rpkg', 54 } 55 new_source_json = json.dumps(new_source_dict) 56 new_source_type = 8 57 session.execute( 58 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 59 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']}) 60 61 fedpkg_build_rows = session.execute( 62 "SELECT * FROM build WHERE source_type=:param", 63 {"param": 7}) 64 65 for build in fedpkg_build_rows: 66 source_dict = json.loads(build['source_json']) if build['source_json'] else {} 67 new_source_dict = { 68 'type': 'git', 69 'clone_url': source_dict.get('clone_url') or '', 70 'committish': source_dict.get('branch') or '', 71 'subdirectory': '', 72 'spec': '', 73 'srpm_build_method': 'rpkg', 74 } 75 new_source_json = json.dumps(new_source_dict) 76 new_source_type = 8 77 session.execute( 78 "UPDATE build SET source_json=:param1, source_type=:param2 WHERE id=:param3", 79 {"param1": new_source_json, "param2": new_source_type, "param3": build['id']})
80 81
82 -def downgrade():
83 pass
84