-
Notifications
You must be signed in to change notification settings - Fork 0
/
AddFieldIndex.py
189 lines (147 loc) · 7.48 KB
/
AddFieldIndex.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
#-------------------------------------------------------------------------------
# Name: module1
# Purpose:
#
# Author: Adolfo.Diaz
#
# Created: 16/04/2018
# Copyright: (c) Adolfo.Diaz 2018
# Licence: <your licence>
#-------------------------------------------------------------------------------
## ===================================================================================
def print_exception():
tb = sys.exc_info()[2]
l = traceback.format_tb(tb)
l.reverse()
tbinfo = "".join(l)
AddMsgAndPrint("\n\n----------ERROR Start-------------------",2)
AddMsgAndPrint("Traceback Info: \n" + tbinfo + "Error Info: \n " + str(sys.exc_type)+ ": " + str(sys.exc_value) + "",2)
AddMsgAndPrint("----------ERROR End-------------------- \n",2)
## ================================================================================================================
def AddMsgAndPrint(msg, severity=0):
# prints message to screen if run as a python script
# Adds tool message to the geoprocessor
#
# Split the message on \n first, so that if it's multiple lines, a GPMessage will be added for each line
print msg
try:
f = open(textFilePath,'a+')
f.write(msg + "\n")
f.close
# Add a geoprocessing message (in case this is run as a tool)
if severity == 0:
arcpy.AddMessage(msg)
elif severity == 1:
arcpy.AddWarning(msg)
elif severity == 2:
arcpy.AddMessage(" ")
arcpy.AddError(msg)
except:
pass
## ===============================================================================================================
def addAttributeIndex(table,fieldList,verbose=True):
""" Attribute indexes can speed up attribute queries on feature classes and tables.
This function adds an attribute index(es) for the fields passed to the table that
is passed in. This function takes in 2 parameters:
1) Table - full path to an existing table or feature class
2) List of fields that exist in table
This function will make sure an existing index is not associated with that field.
Does not return anything."""
try:
# Make sure table exists. - Just in case
if not arcpy.Exists(table):
AddMsgAndPrint("\nAttribute index cannot be created for: " + os.path.basename(table) + " TABLE DOES NOT EXIST",2)
return False
else:
if verbose:
AddMsgAndPrint("\n\tAdding Attribute Indexes to Table: " + os.path.basename(table))
numOfFields = len(fieldList)
arcpy.SetProgressor("step", "Adding Attribute Indexes to Table: " + os.path.basename(table), 0, numOfFields, 1)
# iterate through every field
for fieldToIndex in fieldList:
if verbose:
arcpy.SetProgressorLabel("Adding Attribute Indexes to Table: " + os.path.basename(table) + " - Field: " + fieldToIndex)
# Make sure field exists in table - Just in case
if not len(arcpy.ListFields(table,"*" + fieldToIndex))>0:
if verbose:
AddMsgAndPrint("\t\tAttribute index cannot be created for: " + fieldToIndex + ". FIELD DOES NOT EXIST",2)
arcpy.SetProgressorPosition()
continue
# list of indexes (attribute and spatial) within the table that are
# associated with the field or a field that has the field name in it.
# Important to inspect all associated fields b/c they could be using
# a differently named index
existingIndexes = arcpy.ListIndexes(table,"*" + fieldToIndex)
bFieldIndexExists = False
# check existing indexes to see if fieldToIndex is already associated
# with an index
if existingIndexes > 0:
# iterate through the existing indexes looking for a field match
for index in existingIndexes:
associatedFlds = index.fields
# iterate through the fields associated with existing index.
# Should only be 1 field since multiple fields are not allowed
# in a single FGDB.
for fld in associatedFlds:
# Field is already part of an existing index - Notify User
if fld.name == fieldToIndex:
if verbose:
AddMsgAndPrint("\t\tAttribute Index for " + fieldToIndex + " field already exists",1)
arcpy.SetProgressorPosition()
bFieldIndexExists = True
# Field is already part of an existing index - Proceed to next field
if bFieldIndexExists:
break
# Attribute field index does not exist. Add one.
if not bFieldIndexExists:
newIndex = "IDX_" + fieldToIndex
# UNIQUE setting is not used in FGDBs - comment out
arcpy.AddIndex_management(table,fieldToIndex,newIndex,"#","ASCENDING")
if verbose:
AddMsgAndPrint("\t\tSuccessfully added attribute index for " + fieldToIndex)
arcpy.SetProgressorPosition()
except:
print_exception()
return False
import sys, os, traceback, arcpy
from arcpy import env
if __name__ == '__main__':
try:
<<<<<<< .mine
FGDBpath = r'N:\Pedons\Historical_Pedons.gdb'
||||||| .r32
FGDBpath = r'E:\All_Pedons\NASIS_Pedons\NASIS_Pedons_20180415.gdb'
=======
FGDBpath = r'E:\All_Pedons\NASIS_Pedons\NASIS_Pedons_20180505.gdb'
>>>>>>> .r33
arcpy.env.workspace = FGDBpath
# Add attribute indexes to ALL tables and feature classes in FGDB
fgdbTables = arcpy.ListTables('*')
fgdbFCs = [fgdbTables.append(fc) for fc in arcpy.ListFeatureClasses('*')]
# Contains names of fields that could not get indexed
fieldsNotIndexed = dict()
AddMsgAndPrint("\nAdding Attribute Indexes")
for table in fgdbTables:
tablePath = os.path.join(FGDBpath,table)
fieldNames = []
for field in arcpy.ListFields(tablePath):
# Had to check if text field was too large b/c index was crashing Arc.
# Specifically, have some fields that are 2147483647 in length and causes
# Arc to crash. PASS or Continue doesn't work
if field.type == "String" and field.length > 10000:
if not fieldsNotIndexed.has_key(table):
fieldsNotIndexed.setdefault(table,[field.name])
else:
fieldsNotIndexed[table].append(field.name)
else:
fieldNames.append(field.name)
if not addAttributeIndex(tablePath,fieldNames,True): continue
# Report any fields that did not get indexed
if fieldsNotIndexed:
AddMsgAndPrint("\nThe following fields could not be indexed:",1)
for rec in fieldsNotIndexed.iteritems():
AddMsgAndPrint("\tTable: " + rec[0],1)
for fld in rec[1]:
AddMsgAndPrint("\t\t" + fld)
except:
print_exception()