Files
hudozka/utils/Sync/sync/schedule.py
T

96 lines
2.1 KiB
Python

import os
from glob import glob
from db import db
from sync import Sync, untouched
from utils.fn import lprint, lprint_json, lmap
from utils.hash import hash_file
from utils.io import read_yaml
dir_documents = '/Users/tmshv/Dropbox/Dev/Hud school/Schedules'
class SyncSchedule(Sync):
def __init__(self):
super().__init__()
self.collection = db().schedules
def create_id(self, document):
id = document['id'] if 'id' in document and document['id'] else None
if not id:
id = 'schedule-{period}-{semester}'.format(**document)
document['id'] = id
return document
def create_hash(self, document):
return {
**document,
'hash': hash_file(document['file'])
}
if __name__ == '__main__':
# ALL DOCUMENTS IDS FOUNDED
scope_documents_ids = []
# INIT
os.chdir(dir_documents)
sync = SyncSchedule()
# COLLECT YAML/MD/JPG/PDF MANIFEST FILES
documents = glob('*.yaml')
# READ MANIFEST
documents = lmap(
lambda i: {
'file': i,
**read_yaml(i)
},
documents
)
# # CREATE DOCUMENT IDENTITY
documents = lmap(sync.create_id, documents)
# CREATE DOCUMENT HASH
documents = lmap(sync.create_hash, documents)
# # CREATE SCOPE OF CURRENT SESSION
scope_documents_ids = lmap(
lambda i: i['id'],
documents
)
# # SKIP UNTOUCHED DOCUMENTS
documents = untouched(documents, sync)
# DO HEAVY PROCESS WITH DOCUMENTS
# SYNC TEACHER_PROFILE
documents = lmap(
sync.update,
documents
)
documents_to_remove = sync.query({'id': {'$nin': scope_documents_ids}})
documents_to_remove = lmap(
sync.delete,
map(
lambda i: {'_id': i['_id']},
documents_to_remove
)
)
# SCOPE
print('SCOPE:')
lprint(scope_documents_ids)
# # DELETE
print('DELETE DOCUMENTS:')
lprint_json(documents_to_remove)
# DELETE
print('UPDATE DOCUMENTS:')
lprint_json(documents)
print('[SYNC DOCUMENTS DONE]')