Bridge.py

"""Bridge.py. The main Python-(Swift) plugin bundle entry module"""
import logging
import sys
import objc
from Foundation import NSObject
from Cocoa import NSView
from AppKit import NSGraphicsContext, NSRectToCGRect
import Quartz
from behavior_detector.behavior_detector import BehaviorDetector
from behavior_detector.aggregator import Aggregator
from calendar_integration import quickstart


BridgeInterface = objc.protocolNamed("StackFlow.BridgeInterface")

class Bridge(NSObject, protocols=[BridgeInterface]):
    @classmethod
    def createInstance(self):
        return Bridge.alloc().init()

    def getPythonInformation(self):
        return sys.version

    def addWithA_b_(self, a, b):
        return a + b

    def shouldBreathe(self):
    	return BD.should_breathe()

    def findFlowTime(self):
    	return quickstart.find_flowtime()

    def aggregateByHour(self):
        return AG.aggregate_by_time()

    def aggregateByProgram(self):
        return AG.aggregate_by_program()


logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)

logger.info("Loaded python bundle")

BD = BehaviorDetector()
BD.run()

AG = Aggregator()