first commit

bitable_calendar.py

@@ -0,0 +1,108 @@
+# -*- coding: UTF-8 -*-
+import api
+import config
+import logging
+from datetime import datetime
+import utils
+from mock import schema, raw_data, name_map
+
+LOG_FORMAT = "%(asctime)s - %(levelname)s - %(message)s"
+logging.basicConfig(format=LOG_FORMAT, level=logging.ERROR)
+
+import os
+logging.info(os.getcwd())
+
+def rgb(r: int, g: int, b: int): return (r<<16) + (g<<8) + b
+
+
+def sync_records_to_calendar(client: api.Client, access_token, calendar_id, app_token, table_id):
+    '''generate calendar events from bitable records'''
+
+    records = client.get_records(access_token, app_token, table_id)
+    for record in records:
+        try:
+            fields = record.get('fields')
+            version = fields.get(config.KEY_VERSION)
+            proposal_date = datetime.fromtimestamp(fields.get(config.KEY_STARTUP) / 1000)
+            delivery_date = datetime.fromtimestamp(fields.get(config.KEY_SUBMIT) / 1000)
+            expected_online_time = datetime.fromtimestamp(fields.get(config.KEY_GREY) / 1000)
+            online_time = datetime.fromtimestamp(fields.get(config.KEY_GA) / 1000)
+            print(version, proposal_date, delivery_date, expected_online_time, online_time)
+            print(client.create_event(access_token, calendar_id, start_time=proposal_date, end_time=delivery_date, summary=config.FORMATTER_DEV.format(version), color=rgb(221, 88, 24)))
+            print(client.create_event(access_token, calendar_id, start_time=delivery_date, end_time=expected_online_time, summary=config.FORMATTER_TEST.format(version), color=rgb(85, 220, 101)))
+            print(client.create_event(access_token, calendar_id, start_time=expected_online_time, end_time=online_time, summary=config.FORMATTER_GREY.format(version), color=rgb(118, 244, 244)))
+        except TypeError:
+            pass
+
+
+
+def write_bitable(client: api.Client, access_token, raw_data, schema, name_map) -> str:
+    '''write project release date info to bitable'''
+    table_id = client.create_table(access_token, config.APP_TOKEN, config.TABLE_NAME)
+    resp = client.get_fields_list(access_token, config.APP_TOKEN, table_id)
+    current_fields = resp['items']
+
+    for index, (field_name, field_type) in enumerate(schema.items()):
+        try:
+            if index >= len(current_fields):
+                print(client.add_field(access_token, config.APP_TOKEN, table_id, name_map[field_name], field_type))
+            else:
+                print(client.update_field(
+                    access_token, 
+                    config.APP_TOKEN, 
+                    table_id, 
+                    current_fields[index]['field_id'], 
+                    name_map[field_name],
+                    field_type))
+        except utils.LarkException as e:
+            if e.code == 1254606: # DataNotChange
+                pass
+            else:
+                raise
+
+    resp = client.get_records_list(access_token, config.APP_TOKEN, table_id)
+    current_records = resp['items']
+    if current_records is None:
+        current_records = []
+
+    updated_records = []
+    created_records = []
+
+    for index, version_info in enumerate(raw_data):
+        fields = {}
+        for field_name, field_value in version_info.items():
+            if schema[field_name] == 5:
+                field_value = datetime.strptime(field_value, '%Y-%m-%d').timestamp() * 1000
+            fields[name_map[field_name]] = field_value
+        if index >= len(current_records):
+            created_records.append({'fields': fields})
+        else:
+            updated_records.append({'record_id': current_records[index]['record_id'], 'fields': fields})
+    if updated_records:
+        resp = client.batch_update_records(access_token, config.APP_TOKEN, table_id, updated_records)
+    if created_records:
+        resp = client.batch_create_records(access_token, config.APP_TOKEN, table_id, created_records)
+    return table_id
+
+def bitable_to_calendar():
+    '''write project release date info to bitable and sync it to calendar'''
+    # init api client
+    client = api.Client(config.LARK_HOST)
+
+    # get tenant access token        
+    access_token = client.get_tenant_access_token(config.APP_ID, config.APP_SECRET)
+
+    # create a new table and push local json-like version iteration data to it
+    table_id = write_bitable(client, access_token, raw_data, schema, name_map)
+
+    # create new calendar and get its id
+    calendar = client.create_calendar(access_token, permissions='public', summary=config.SUMMARY)
+    print(calendar)
+    calendar_id = calendar['calendar_id']
+
+    # sync records to calendar
+    sync_records_to_calendar(client, access_token, calendar_id, config.APP_TOKEN, table_id)
+
+
+if __name__ == "__main__":
+    bitable_to_calendar()