carnotpy 0.1__tar.gz

carnotpy-0.1/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.1
+Name: carnotpy
+Version: 0.1
+Summary: Carnot Connection as a Python Package
+Home-page: UNKNOWN
+Author: Carnot Innovations
+License: UNKNOWN
+Platform: UNKNOWN
+
+UNKNOWN
+

carnotpy-0.1/README.md

@@ -0,0 +1,2 @@
+# carnotpy
+Carnot Connection as a Python Package

carnotpy-0.1/carnotpy.egg-info/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.1
+Name: carnotpy
+Version: 0.1
+Summary: Carnot Connection as a Python Package
+Home-page: UNKNOWN
+Author: Carnot Innovations
+License: UNKNOWN
+Platform: UNKNOWN
+
+UNKNOWN
+

carnotpy-0.1/carnotpy.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+README.md
+carnotpy.py
+setup.py
+carnotpy.egg-info/PKG-INFO
+carnotpy.egg-info/SOURCES.txt
+carnotpy.egg-info/dependency_links.txt
+carnotpy.egg-info/requires.txt
+carnotpy.egg-info/top_level.txt

carnotpy-0.1/carnotpy.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

carnotpy-0.1/carnotpy.egg-info/requires.txt

@@ -0,0 +1,3 @@
+pandas
+requests
+python-dateutil

carnotpy-0.1/carnotpy.egg-info/top_level.txt

@@ -0,0 +1 @@
+carnotpy

carnotpy-0.1/carnotpy.py

@@ -0,0 +1,230 @@
+import pandas as pd
+import datetime
+import requests
+from dateutil.relativedelta import relativedelta
+
+
+
+def process_equipment(payload, equipment_name):
+    if payload['filters'].get('equipment_name'):
+        if payload.get('pipeline') and len(payload['pipeline']) > 0:
+            for stage in payload['pipeline']:
+                if stage['type'] == 'filter':
+                    stage['filter_by'] = stage['filter_by'].replace(payload['filters']['equipment_name'] + ' - ',
+                                                                    equipment_name + ' - ')
+                elif stage.get('formula') and stage['formula'] != '':
+                    stage['formula'] = stage['formula'].replace(payload['filters']['equipment_name'] + ' - ',
+                                                                equipment_name + ' - ')
+                elif stage['type'] == 'insert_series':
+                    stage['name'] = stage['name'].replace(payload['filters']['equipment_name'] + ' - ',
+                                                          equipment_name + ' - ')
+                    if stage.get('filters') and stage['filters']['equipment_name'] == payload['filters'][
+                        'equipment_name']:
+                        stage['filters']['equipment_name'] = equipment_name
+        payload['filters']['equipment_name'] = equipment_name
+
+
+def process_date_range(payload, date_range):
+    if payload['useCustomRange']:
+        if payload['useCustomRelativeRange']:
+            rel_range = payload.get('customRelativeRange', '30 days')
+            new_range = None
+            base_date = datetime.datetime.now()
+            start_date = base_date - datetime.timedelta(days=30)
+            if rel_range == 'current week':
+                start_date = base_date - datetime.timedelta(days=base_date.weekday())
+            elif rel_range == 'current month':
+                start_date = base_date.replace(day=1)
+            elif rel_range == 'today':
+                start_date = base_date.replace(hour=0, minute=0, second=0, microsecond=0)
+            elif rel_range == 'current year':
+                start_date = base_date.replace(month=1, day=1)
+            elif rel_range == 'last week':
+                base_date = base_date - datetime.timedelta(weeks=1)
+                end_date = base_date.replace(hour=23, minute=59, second=59, microsecond=999999)
+                start_date = end_date - datetime.timedelta(days=end_date.weekday())
+            elif rel_range == 'last month':
+                base_date = base_date.replace(day=1) - datetime.timedelta(days=1)
+                end_date = base_date.replace(hour=23, minute=59, second=59, microsecond=999999)
+                start_date = end_date.replace(day=1)
+            elif rel_range == 'yesterday':
+                base_date = base_date - datetime.timedelta(days=1)
+                end_date = base_date.replace(hour=23, minute=59, second=59, microsecond=999999)
+                start_date = end_date.replace(hour=0, minute=0, second=0, microsecond=0)
+            elif rel_range == 'last year':
+                base_date = base_date.replace(year=base_date.year - 1, month=12, day=31)
+                end_date = base_date.replace(hour=23, minute=59, second=59, microsecond=999999)
+                start_date = end_date.replace(year=base_date.year - 1, month=1, day=1)
+
+            if rel_range in ['yesterday', 'today']:
+                new_range = {
+                    'start': start_date.strftime('%Y-%m-%d %H:%M:%S'),
+                    'end': base_date.strftime('%Y-%m-%d %H:%M:%S')
+                }
+            else:
+                new_range = {
+                    'start': start_date.strftime('%Y-%m-%d'),
+                    'end': (base_date.strftime('%Y-%m-%d %H:%M:%S') if len(
+                        base_date.strftime('%Y-%m-%d')) < 12 else base_date.strftime('%Y-%m-%d'))
+                }
+
+            payload['date_range'] = [new_range['start'], new_range['end']]
+
+        elif payload.get('customRange'):
+            if payload['customRangeStartOnly'] and date_range and date_range['end']:
+                payload['date_range'] = [payload['customRange']['start'], (
+                    date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])]
+            else:
+                payload['date_range'] = [payload['customRange']['start'], (
+                    payload['customRange']['end'] + ' 23:59:59' if len(payload['customRange']['end']) < 12 else
+                    payload['customRange']['end'])]
+
+    elif date_range:
+        if payload.get('offsetRange') and payload.get('offsetType'):
+            if payload['offsetRangeStartOnly']:
+                if payload['offsetType'] in ['year', 'years'] and payload.get('offsetPreciseYear'):
+                    payload['date_range'] = [
+                        datetime.datetime.strptime(date_range['start'], '%Y-%m-%d').replace(
+                            year=payload['offsetPreciseYear']).strftime('%Y-%m-%d'),
+                        (date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])
+                    ]
+                elif payload['offsetType'] in ['month', 'months']:
+                    offset = relativedelta(months=-payload['offsetAmount'])
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') + offset).strftime('%Y-%m-%d'),
+                        (date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])
+                    ]
+                elif payload['offsetType'] in ['year', 'years']:
+                    offset = relativedelta(years=-payload['offsetAmount'])
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') + offset).strftime('%Y-%m-%d'),
+                        (date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])
+                    ]
+                else:
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') - datetime.timedelta(
+                            **{payload['offsetType']: payload['offsetAmount']})).strftime('%Y-%m-%d'),
+                        (date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])
+                    ]
+            else:
+                if payload['offsetType'] in ['year', 'years'] and payload.get('offsetPreciseYear'):
+                    payload['date_range'] = [
+                        datetime.datetime.strptime(date_range['start'], '%Y-%m-%d').replace(
+                            year=payload['offsetPreciseYear']).strftime('%Y-%m-%d'),
+                        datetime.datetime.strptime(date_range['end'], '%Y-%m-%d').replace(
+                            year=payload['offsetPreciseYear']).strftime('%Y-%m-%d') + ' 23:59:59'
+                    ]
+                elif payload['offsetType'] in ['month', 'months']:
+                    offset = relativedelta(months=-payload['offsetAmount'])
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') + offset).strftime('%Y-%m-%d'),
+                        (datetime.datetime.strptime(date_range['end'], '%Y-%m-%d') + offset).strftime(
+                            '%Y-%m-%d') + ' 23:59:59'
+                    ]
+                elif payload['offsetType'] in ['year', 'years']:
+                    offset = relativedelta(years=-payload['offsetAmount'])
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') + offset).strftime('%Y-%m-%d'),
+                        (datetime.datetime.strptime(date_range['end'], '%Y-%m-%d') + offset).strftime(
+                            '%Y-%m-%d') + ' 23:59:59'
+                    ]
+                else:
+                    payload['date_range'] = [
+                        (datetime.datetime.strptime(date_range['start'], '%Y-%m-%d') - datetime.timedelta(
+                            **{payload['offsetType']: payload['offsetAmount']})).strftime('%Y-%m-%d'),
+                        (datetime.datetime.strptime(date_range['end'], '%Y-%m-%d') - datetime.timedelta(
+                            **{payload['offsetType']: payload['offsetAmount']})).strftime('%Y-%m-%d') + ' 23:59:59'
+                    ]
+        else:
+            payload['date_range'] = [date_range['start'], (
+                date_range['end'] + ' 23:59:59' if len(date_range['end']) < 12 else date_range['end'])]
+
+
+class CarnotPy:
+
+    def __init__(self, url, username, password):
+        self.url = url
+        self.username = username
+        self.password = password
+
+    def get_token(self):
+        result = requests.post(self.url + "/login/", json={"user_name": self.username, "password": self.password})
+        assert result.status_code == 200
+        return result.json()["token"]
+
+    def discover(self):
+        token = self.get_token()
+        result = requests.post(self.url + "/point-discovery/",
+                               headers={"Authorization": "Bearer " + token}
+                               )
+        assert result.status_code == 200
+        return result.json()["result"]
+
+    def read_history(self, sensor_ids, start_time, end_time):
+        token = self.get_token()
+        result = requests.post(
+            self.url + "/historical-data/",
+            json={
+                "date_range": [start_time, end_time],
+                "sensor_ids": sensor_ids
+            },
+            headers={"Authorization": "Bearer " + token, 'Content-Type': 'application/json'}
+        )
+        assert result.status_code == 200
+        data = result.json()
+        data['timestamp'] = pd.to_datetime(data['timestamp'])
+        df = pd.DataFrame(data)
+        df = df.set_index('timestamp')
+        return df
+
+    def read_data3(self, payload, token=None):
+        if token is None:
+            token = self.get_token()
+        result = requests.post(
+            self.url + "/data3/",
+            json=payload,
+            headers={"Authorization": "Bearer " + token, 'Content-Type': 'application/json'}
+        )
+        assert result.status_code == 200
+        data = result.json()
+        data['timestamp'] = pd.to_datetime(data['timestamp'])
+        df = pd.DataFrame(data)
+        df = df.set_index('timestamp')
+        return df
+
+    def read_data3_query(self, query, facility_name, date_range, equipment=None, on_only=False, parameters=None):
+        token = self.get_token()
+        payload_array = query['data-payload']
+        df = None
+        index = None
+        for payload in payload_array:
+            payload['parameters'] = parameters
+            process_date_range(payload, date_range)
+            if 'facility_name' not in payload["filters"]:
+                payload["filters"]["facility_name"] = facility_name
+            if equipment is not None:
+                process_equipment(payload, equipment)
+            if on_only or payload.get('onOnly', False):
+                if 'pipeline' not in payload:
+                    payload['pipeline'] = []
+                payload['pipeline'].append({
+                                  "type": "filter",
+                                  "filter_op": "eq",
+                                  "filter_by": "* - on",
+                                  "filter_value": 1,
+                              })
+            del(payload["seriesOverrides"])
+            new_df = self.read_data3(payload, token=token)
+            if index is None:
+                index = new_df.index
+            new_df.index = range(len(new_df))
+            if df is None:
+                df = new_df
+            else:
+                df = df.join(new_df)
+
+        df.index = index
+        return df
+
+
+

carnotpy-0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

carnotpy-0.1/setup.py

@@ -0,0 +1,14 @@
+from setuptools import setup
+
+setup(
+    name='carnotpy',
+    version='0.1',
+    author='Carnot Innovations',
+    description='Carnot Connection as a Python Package',
+    py_modules=['carnotpy'],
+    install_requires=[
+        'pandas',
+        'requests',
+        'python-dateutil',
+    ],
+)