#!/usr/bin/env python3

# external modules

import openpyxl

from openpyxl.styles import Font, Alignment

from openpyxl.styles.colors import BLUE

# my modules

import DCC

import Config as CF

import FileSys

import MyUtil

# Set Excel Styles

# Excel hyperlink style is calibri 11, underline blue

font_url_style = Font(color = BLUE, underline = 'single')

bold_style = Font(bold = True)

align_hv_cen_style = Alignment(horizontal = 'center', vertical = 'center')

align_ver_cen_style = Alignment(vertical = 'center')

align_hv_cen_wrap_style = Alignment(horizontal = 'center', vertical = 'center', wrap_text = True)

#function uses kwarg argument to check a load flag is given. If a flag is given then the function will load the existing file. It gives the option to load from the disk or recreate the json file

def return_tree(s, target, rootfilename, **kwargs):

return(tr)

def flat_tree(tree, key, list):

return(list)

#function that iterates through the tree and prints document information

def iter_print_tree(s, tree, key, indent):

iter_print_tree(s, tree, col, indent+' ')

# This function calls the iterative version of print tree

def print_tree(s,tree):

iter_print_tree(s, tree, 'root', '')

# creates a hyperlink using the url given in arguments

def href_str(name, url):

return('<a href="'+url+'">'+name+'</a>')

def url_view(handle):

return('https://docushare.tmt.org/docushare/dsweb/ServicesLib/' + handle + '/View')

def url_access(handle):

return('https://docushare.tmt.org/docushare/dsweb/View/' + handle)

def url_perm(handle):

return('https://docushare.tmt.org/docushare/dsweb/ServicesLib/' + handle + '/Permissions')

def url_ver(handle):

return('https://docushare.tmt.org/docushare/dsweb/ServicesLib/' + handle + '/History')

def url_loc(handle):

return('https://docushare.tmt.org/docushare/dsweb/ServicesLib/' + handle + '/Location')

# function that iterates through html tree

def iter_html_tree(s, htmlfile, tree, key, indent, **kwargs):

iter_html_tree(s, htmlfile, tree, col, indent+2, **kwargs)

def html_tree(s,tree,froot,**kwargs):

htmlfile.close

# function sets excel workbook headings for html tree

def xls_tree_headings(ws):

ws.column_dimensions["J"].width = 35.0

# Prints data to xls document

def xls_print_ssrow(ws, collData, docData, ssrow):

ws.cell(row = ssrow, column = col).alignment = align_hv_cen_style

# Global Variable ssrow

ssrow = 2

#iterates through xls version of html tree

def xls_tree_iter(s,ws,tree,col, **kwargs):

xls_tree_iter(s,ws,tree,newcol,**kwargs)

def xls_tree(s,tree,col,fname, **kwargs):

wb.save(CF.reportfilepath + fname + '.xls')

# Funciton builds html tree and uses prop_get to get information of the target collection

# Uses kwarg argument to recognize exclude list

def build_tree(s, keyname, target, tree, **kwargs):

return(tree)

def get_tree(s, collhandle, **kwargs):

return(build_tree(s, collhandle, collhandle, build_root(collhandle), **kwargs))

def get_flat_tree(tree):

fl = flat_tree(tree, 'root', [])

return(fl)

def build_root(collhandle):

tree = {}

tree['root'] = {'collections' : [collhandle], 'documents' : [], 'others' : []}

return(tree)

def test_tree():

print(fl)

if __name__ == '__main__':

print("Running module test code for",__file__)

# test_tree()

# Login to DCC

s = DCC.login(CF.dcc_url + CF.dcc_login)

# froot = 'Listing of Scott Collection'

# coll = 'Collection-286'

# froot = 'Listing M1CS Preliminary Design Review Collections'

# coll = 'Collection-10725'

#

# tr = return_tree(s, coll, froot)

# # print_tree(s,tr)

# html_tree(s,tr,froot)

# load_flag = True

# froot = 'Listing of WFOS for Suijian'

# coll = 'Collection-7798'

# tr = return_tree(s, coll, froot)

# html_tree(s,tr,froot)

froot = 'Test of HTML bug'

coll = 'Collection-2656'

tr = return_tree(s, coll, froot)

html_tree(s,tr,froot)

# froot = 'Listing of IRIS'

# coll = 'Collection-2463'

# froot = 'Listing of STR CID'

# coll = 'Collection-10669'

# coll = 'Collection-11377'

# tr = return_tree(s, coll, froot)

# xls_tree(s,tr,coll,froot)

# html_tree(s,tr,froot)

# froot = 'Listing of STR CID'

# coll = 'Collection-10668'

# tr = return_tree(s, coll, froot)

# html_tree(s,tr,froot)

# froot = 'Listing of 2013 [Nov 12-14] Telescope Structure System Preliminary Design Review'

# coll = 'Collection-7095'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2014 [April 15 -16] Telescope Structure System Controls-Software Preliminary Design Review'

# coll = 'Collection-7857'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2014 [April 15 -16] Telescope Structure System Controls-Software Preliminary Design Review'

# coll = 'Collection-7857'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2014 [Nov 18-20] Segment Handling System Preliminary Design Review'

# coll = 'Collection-8674'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2015 [Feb 17-20] Telescope Structure (STR) Final Design Review Presentation (FDRP1)'

# coll = 'Collection-9067'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2015 [Jul 27-29] Telescope Structure (STR) Final Design Review Presentation (FDRP2)'

# coll = 'Collection-10071'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)

#

# froot = 'Listing of 2015 [Oct 08-09] Telescope Structure (STR) Long Lead Procurement Review (LPR)'

# coll = 'Collection-11045'

# tr = return_tree(s, coll, froot, Load = load_flag)

# html_tree(s,tr,froot)