examples/simple-make/count-lines-3

#!/usr/bin/env python
#
# This source file is part of the Swift.org open source project
#
# Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
# Licensed under Apache License v2.0 with Runtime Library Exception
#
# See http://swift.org/LICENSE.txt for license information
# See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors

import os

import util
from simplebuild import *

class GetFileInfoRule(SimpleAsyncRule):
    """
    Rule which synchronizes the state of external file.
    """

    def __init__(self, path):
        self.path = str(path)

    def run(self):
        util.note("getting file info: %r" % (self.path,))
        return util.get_stat_info(self.path)

    def is_data_valid(self, engine, data):
        # Check if the current info is up-to-date.
        util.note("checking file info: %r" % (self.path,))
        return util.get_stat_info(self.path) == data
        
class GetDirectoryContentsRule(SimpleAsyncRule):
    def __init__(self, path):
        self.path = path

    def start(self, engine):
        # Get the file info for the directory, to ensure we treat the external
        # directory as an input.
        engine.task_needs_input(self, GetFileInfoRule.asKey(self.path))

    def provide_data(self, engine, input_id, data):
        pass

    def run(self):
        util.note("listing directory contents: %r" % (str(self.path),))
        try:
            return sorted(os.listdir(self.path))
        except OSError:
            return []
    
class CountSourceLinesRule(SimpleAsyncRule):
    def __init__(self, path):
        self.path = str(path)
        self.result = None

    def run(self):
        with open(self.path) as file:
            return len(file.readlines())

    def start(self, engine):
        engine.task_needs_input(self, GetDirectoryContentsRule.asKey(self.path))

    def provide_data(self, engine, input_id, data):
        # The input data will be the list of files.
        self.result = 0
        for input in data:
            if input.endswith(".c"):
                with open(os.path.join(self.path, input)) as file:
                    self.result += len(file.readlines())

    def run(self):
        return self.result

# Get the input path.
_,path = sys.argv

# Run the job
engine = DataDrivenEngine(globals())
engine.attach_db(".count-source-lines.db")

result = engine.build(CountSourceLinesRule.asKey(path))
print result