| #!/usr/bin/env python3 |
| # |
| # Copyright (C) 2016 Codethink Limited |
| # |
| # This program is free software; you can redistribute it and/or |
| # modify it under the terms of the GNU Lesser General Public |
| # License as published by the Free Software Foundation; either |
| # version 2 of the License, or (at your option) any later version. |
| # |
| # This library is distributed in the hope that it will be useful, |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| # Lesser General Public License for more details. |
| # |
| # You should have received a copy of the GNU Lesser General Public |
| # License along with this library. If not, see <http://www.gnu.org/licenses/>. |
| # |
| # Authors: |
| # Tristan Van Berkom <tristan.vanberkom@codethink.co.uk> |
| # Jürg Billeter <juerg.billeter@codethink.co.uk> |
| |
| # System imports |
| import os |
| |
| # BuildStream toplevel imports |
| from .. import Consistency |
| from ..plugin import _plugin_lookup |
| from .. import _yaml |
| |
| # Local imports |
| from . import Queue, QueueType |
| |
| |
| # A queue which tracks sources |
| # |
| class TrackQueue(Queue): |
| |
| action_name = "Track" |
| complete_name = "Tracked" |
| queue_type = QueueType.FETCH |
| |
| def __init__(self): |
| super(TrackQueue, self).__init__() |
| |
| def process(self, element): |
| return element._track() |
| |
| def skip(self, element): |
| # We can skip elements entirely if they have no sources. |
| return len(list(element.sources())) == 0 |
| |
| def done(self, element, result, returncode): |
| |
| if returncode != 0: |
| return False |
| |
| changed = False |
| |
| # Set the new refs in the main process one by one as they complete |
| for unique_id, new_ref in result: |
| source = _plugin_lookup(unique_id) |
| if source._set_ref(new_ref, source._Source__origin_node): |
| |
| changed = True |
| project = source.get_project() |
| toplevel = source._Source__origin_toplevel |
| filename = source._Source__origin_filename |
| fullname = os.path.join(project.element_path, filename) |
| |
| # Here we are in master process, what to do if writing |
| # to the disk fails for some reason ? |
| try: |
| _yaml.dump(toplevel, fullname) |
| except OSError as e: |
| source.error("Failed to update project file", |
| detail="{}: Failed to rewrite tracked source to file {}: {}" |
| .format(source, fullname, e)) |
| |
| # Forcefully recalculate the element's consistency state after successfully |
| # tracking, this is avoid a following fetch queue operating on the sources |
| # if the tracked ref is cached as a result. |
| # |
| context = element.get_context() |
| context._push_message_depth(True) |
| element._consistency(recalculate=True) |
| context._pop_message_depth() |
| |
| # We'll appear as a skipped element if tracking resulted in no change |
| return changed |