aboutsummaryrefslogtreecommitdiff
path: root/android_bench_suite/discard_patches.py
blob: 7a4b0b561d8de2a9e9afa07077606f18614a59f1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/usr/bin/env python2
#
# Copyright 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Script to discard all the patches added to Android for this suite"""

from __future__ import print_function

import config
import os
import subprocess


def discard_git(path):
  try:
    subprocess.check_call(['git', '-C', path, 'reset'])
    subprocess.check_call(['git', '-C', path, 'clean', '-fdx'])
    subprocess.check_call(['git', '-C', path, 'stash'])
    print('Patch in %s removed successfully!' % path)
  except subprocess.CalledProcessError:
    print('Error while removing patch in %s' % path)


def dispatch_skia():
  skia_dir = os.path.join(config.android_home, config.bench_dict['Skia'])
  discard_git(skia_dir)


def dispatch_autotest():
  autotest_dir = os.path.join(config.android_home, config.autotest_dir)
  discard_git(autotest_dir)


def dispatch_panorama():
  panorama_dir = os.path.join(config.android_home,
                              config.bench_dict['Panorama'])
  discard_git(panorama_dir)


def dispatch_synthmark():
  synthmark_dir = 'synthmark'
  try:
    subprocess.check_call(
        ['rm', '-rf',
         os.path.join(config.android_home, synthmark_dir)])
    print('Synthmark patch removed successfully!')
  except subprocess.CalledProcessError:
    print('Synthmark is not removed. Error occurred.')


def main():
  dispatch_skia()
  dispatch_autotest()
  dispatch_panorama()
  dispatch_synthmark()


if __name__ == '__main__':
  main()