Python-Projects/FieldValueCompare.py at master · andersolson/Python-Projects · GitHub

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
'''****************************************************************
FieldValueCompare.py
Author(s): Anders Olson
Usage: Run script using python IDE or similar
Description:
        Script traverses ID field for 2 shapefiles and compares the
        list of IDs to see what is missing.
****************************************************************'''

import arcpy
from collections import Counter

##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##
#================================#
# Configure logger
#================================#
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##

def outputMessage(msg):
    print(msg)
    arcpy.AddMessage(msg)

def outputError(msg):
    print(msg)
    arcpy.AddError(msg)

##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##
#================================#
# Define environment and messaging
#================================#
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##

# Set the overwriteOutput ON
arcpy.gp.overwriteOutput = True

# Set workspace to be in memory for faster run time
arcpy.env.workspace = "in_memory"

outputMessage("Workspace is: {}".format(arcpy.env.workspace))

##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##
#================================#
# Define Functions
#================================#
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##

def createValueList(shpIn,fld):
    # List to return all values for given field
    fieldValues = []

    # Populate a list of all IDs found in the desired ID field
    for row in arcpy.da.SearchCursor(shpIn, [fld]):
        fieldValues.append(row[0])

    # # Print a list of all ids and no repeating ids i.e. 'keys'
    # print(Counter(uniqueValues).keys())
    #
    # # Print a count for occurrence of each id found in original list i.e. 'values'
    # print(Counter(uniqueValues).values())

    # Output a list of all values found in the field
    return(fieldValues)

def diffLST(inLST1,inLST2):
    return (list(set(inLST1) - set(inLST2)))

##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##
#================================#
# Define variables
#================================#
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##

# Input shapefiles or dbf with id field to compare
shp1 = r'C:\Users\is_olson\Documents\Projects\Central-Square\reports\CSUpdateReport_111825.shp'
shp2 = r'C:\Users\is_olson\Documents\Projects\Central-Square\reports\CSUpdateReport_111825_TEST.shp'

# Field name that the value search will use to find all values
fieldName = 'PIN'

##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##
#================================#
# Call Functions
#================================#
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~##

# Get list of all values in given field from each shapefile to compare
shp1Values = createValueList(shp1,fieldName)
shp2Values = createValueList(shp2,fieldName)

# Get list of values that do not match
diff = diffLST(shp1Values,shp2Values)
# diff = diffLST(shp2Values,shp1Values)

# print(shp1Values)
# print(shp2Values)
print(diff)