Script 512ff2b9eb6c_unify_scm_source_types_for_packages_py
[hide private]
[frames] | no frames]

Source Code for Script script-512ff2b9eb6c_unify_scm_source_types_for_packages_py

 1  """unify scm source types for packages 
 2   
 3  Revision ID: 512ff2b9eb6c 
 4  Revises: a8654abcec65 
 5  Create Date: 2017-10-14 13:24:44.164388 
 6   
 7  """ 
 8   
 9  # revision identifiers, used by Alembic. 
10  revision = '512ff2b9eb6c' 
11  down_revision = 'a8654abcec65' 
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_package_rows = session.execute( 22 "SELECT * FROM package WHERE source_type=:param", 23 {"param": 3}) 24 25 for package in tito_package_rows: 26 source_dict = json.loads(package['source_json']) if package['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 package SET source_json=:param1, source_type=:param2 WHERE id=:param3", 39 {"param1": new_source_json, "param2": new_source_type, "param3": package['id']}) 40 41 mock_package_rows = session.execute( 42 "SELECT * FROM package WHERE source_type=:param", 43 {"param": 4}) 44 45 for package in mock_package_rows: 46 source_dict = json.loads(package['source_json']) if package['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 package SET source_json=:param1, source_type=:param2 WHERE id=:param3", 59 {"param1": new_source_json, "param2": new_source_type, "param3": package['id']}) 60 61 fedpkg_package_rows = session.execute( 62 "SELECT * FROM package WHERE source_type=:param", 63 {"param": 7}) 64 65 for package in fedpkg_package_rows: 66 source_dict = json.loads(package['source_json']) if package['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 package SET source_json=:param1, source_type=:param2 WHERE id=:param3", 79 {"param1": new_source_json, "param2": new_source_type, "param3": package['id']})
80 81
82 -def downgrade():
83 pass
84