Compare commits
3 Commits
54e188bc55
...
6bbf29493c
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6bbf29493c | ||
|
|
c5708d6d79 | ||
|
|
2e17eac57d |
105
README.md
105
README.md
@ -1,2 +1,105 @@
|
|||||||
# 753-Data-Sync
|
# 753 Data Sync
|
||||||
|
|
||||||
|
This script fetches enforcement data from an external API, truncates a specified feature layer in ArcGIS, and adds the fetched data as features to the layer. The script performs the following tasks:
|
||||||
|
|
||||||
|
1. **Truncate the specified layer** in ArcGIS to clear any previous features before adding new ones.
|
||||||
|
2. **Fetch data** from an API in paginated form.
|
||||||
|
3. **Save data** from each API response to individual JSON files.
|
||||||
|
4. **Aggregate all data** from all pages into one JSON file.
|
||||||
|
5. **Add the aggregated data** as features to an ArcGIS feature service.
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
|
||||||
|
- Python 3.6 or higher
|
||||||
|
- Required Python packages (see `requirements.txt`)
|
||||||
|
- ArcGIS Online credentials (username and password)
|
||||||
|
- `.env` file for configuration (see below for details)
|
||||||
|
|
||||||
|
### Install dependencies
|
||||||
|
|
||||||
|
You can install the required dependencies using `pip`:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
pip install -r requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
## Configuration
|
||||||
|
|
||||||
|
Before running the script, you'll need to configure some environment variables. Create a `.env` file with the following details:
|
||||||
|
|
||||||
|
```env
|
||||||
|
API_URL=your_api_url
|
||||||
|
AGOL_USER=your_arcgis_online_username
|
||||||
|
AGOL_PASSWORD=your_arcgis_online_password
|
||||||
|
HOSTNAME=your_arcgis_host
|
||||||
|
INSTANCE=your_arcgis_instance
|
||||||
|
FS=your_feature_service
|
||||||
|
LAYER=your_layer_id
|
||||||
|
```
|
||||||
|
|
||||||
|
### Variables
|
||||||
|
|
||||||
|
- **API_URL**: The URL of the API you are fetching data from.
|
||||||
|
- **AGOL_USER**: Your ArcGIS Online username.
|
||||||
|
- **AGOL_PASSWORD**: Your ArcGIS Online password.
|
||||||
|
- **HOSTNAME**: The hostname of your ArcGIS Online instance (e.g., `www.arcgis.com`).
|
||||||
|
- **INSTANCE**: The instance name of your ArcGIS Online service.
|
||||||
|
- **FS**: The name of the feature service you are working with.
|
||||||
|
- **LAYER**: The ID or name of the layer to truncate and add features to.
|
||||||
|
|
||||||
|
## Script Usage
|
||||||
|
|
||||||
|
You can run the script with the following command:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python 753DataSync.py --results_per_page <number_of_results_per_page>
|
||||||
|
```
|
||||||
|
|
||||||
|
### Arguments
|
||||||
|
|
||||||
|
- **--results_per_page** (optional): The number of results to fetch per page (default: 100).
|
||||||
|
|
||||||
|
## Functionality
|
||||||
|
|
||||||
|
1. **Truncate Layer**: Before fetching and adding any new data, the script will call the `truncate` function to clear out any existing features from the specified layer. This ensures that the feature layer is empty and ready for the new data.
|
||||||
|
|
||||||
|
2. **Fetch Data**: The script will then fetch data from the specified API in pages. Each page is fetched sequentially until all data is retrieved.
|
||||||
|
|
||||||
|
3. **Save Data**: Data from each page will be saved to an individual JSON file, with the filename including the page number and timestamp. The aggregated data (all pages combined) is saved to a separate file.
|
||||||
|
|
||||||
|
4. **Add Features**: After all the data has been fetched and saved, the script will send the aggregated data as features to the specified ArcGIS feature layer.
|
||||||
|
|
||||||
|
### Example Output
|
||||||
|
|
||||||
|
- Individual page files are saved in the `data/` directory with filenames like `enforcement_page_1_results_100_2025-03-26_14-30-45.json`.
|
||||||
|
- The aggregated file is saved as `aggregated_enforcement_results_2025-03-26_14-30-45.json`.
|
||||||
|
|
||||||
|
Logs will also be generated in the `753DataSync.log` file and printed to the console.
|
||||||
|
|
||||||
|
## Error Handling
|
||||||
|
|
||||||
|
- If an error occurs while fetching data, the script will log the error and stop execution.
|
||||||
|
- If the `truncate` or `add_features` operations fail, the script will log the error and stop execution.
|
||||||
|
- The script handles HTTP errors and network-related errors gracefully.
|
||||||
|
|
||||||
|
## Example Output (Log)
|
||||||
|
|
||||||
|
```
|
||||||
|
2025-03-26 14:30:45 - INFO - Attempting to truncate layer on https://www.arcgis.com/...
|
||||||
|
2025-03-26 14:30:50 - INFO - Successfully truncated layer: https://www.arcgis.com/...
|
||||||
|
2025-03-26 14:30:51 - INFO - Making request to: https://api.example.com/1/100
|
||||||
|
2025-03-26 14:30:55 - INFO - Data saved to data/enforcement_page_1_results_100_2025-03-26_14-30-45.json
|
||||||
|
2025-03-26 14:30:56 - INFO - No more data to fetch, stopping pagination.
|
||||||
|
2025-03-26 14:30:57 - INFO - Data saved to data/aggregated_enforcement_results_2025-03-26_14-30-45.json
|
||||||
|
2025-03-26 14:31:00 - INFO - Features added successfully.
|
||||||
|
```
|
||||||
|
|
||||||
|
## Troubleshooting
|
||||||
|
|
||||||
|
- If the script stops unexpectedly, check the logs (`753DataSync.log`) for detailed error information.
|
||||||
|
- Ensure your `.env` file is correctly configured with valid credentials and API URL.
|
||||||
|
- Make sure the specified ArcGIS layer is accessible and has the correct permissions for truncating and adding features.
|
||||||
|
|
||||||
|
## License
|
||||||
|
|
||||||
|
This project is licensed under the **GNU General Public License v3.0** or later - see the [LICENSE](LICENSE) file for details.
|
||||||
231
app.py
Normal file
231
app.py
Normal file
@ -0,0 +1,231 @@
|
|||||||
|
import requests
|
||||||
|
import logging
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import json
|
||||||
|
from datetime import datetime
|
||||||
|
import argparse
|
||||||
|
import urllib.parse
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
# Configuration
|
||||||
|
BASE_URL = "{}/{}/{}"
|
||||||
|
|
||||||
|
# Setup logging
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel(logging.INFO)
|
||||||
|
|
||||||
|
# File handler
|
||||||
|
file_handler = logging.FileHandler('753DataSync.log')
|
||||||
|
file_handler.setLevel(logging.INFO)
|
||||||
|
|
||||||
|
# Stream handler (console output)
|
||||||
|
stream_handler = logging.StreamHandler(sys.stdout)
|
||||||
|
stream_handler.setLevel(logging.INFO)
|
||||||
|
|
||||||
|
# Log format
|
||||||
|
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
|
||||||
|
file_handler.setFormatter(formatter)
|
||||||
|
stream_handler.setFormatter(formatter)
|
||||||
|
|
||||||
|
# Add handlers to the logger
|
||||||
|
logger.addHandler(file_handler)
|
||||||
|
logger.addHandler(stream_handler)
|
||||||
|
|
||||||
|
def fetch_data(api_url, page_number, results_per_page):
|
||||||
|
"""Fetches data from the API and returns the response."""
|
||||||
|
url = BASE_URL.format(api_url, page_number, results_per_page)
|
||||||
|
|
||||||
|
try:
|
||||||
|
logger.info(f"Making request to: {url}")
|
||||||
|
response = requests.get(url)
|
||||||
|
|
||||||
|
# Check for HTTP errors
|
||||||
|
response.raise_for_status()
|
||||||
|
|
||||||
|
# Return JSON data
|
||||||
|
return response.json()
|
||||||
|
|
||||||
|
except requests.exceptions.HTTPError as http_err:
|
||||||
|
logger.error(f"HTTP error occurred: {http_err}")
|
||||||
|
sys.exit(1)
|
||||||
|
except requests.exceptions.RequestException as req_err:
|
||||||
|
logger.error(f"Request error occurred: {req_err}")
|
||||||
|
sys.exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
logger.error(f"An unexpected error occurred: {err}")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
def save_json(data, filename):
|
||||||
|
"""Saves JSON data to a file."""
|
||||||
|
try:
|
||||||
|
# Ensure directory exists
|
||||||
|
if not os.path.exists('data'):
|
||||||
|
os.makedirs('data')
|
||||||
|
|
||||||
|
# Save data to file
|
||||||
|
with open(filename, 'w', encoding='utf-8') as f:
|
||||||
|
json.dump(data, f, ensure_ascii=False, indent=4)
|
||||||
|
|
||||||
|
logger.info(f"Data saved to {filename}")
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error saving JSON data: {e}")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
def parse_arguments():
|
||||||
|
"""Parse command-line arguments."""
|
||||||
|
parser = argparse.ArgumentParser(description="Fetch enforcement data from the NYSDPS API.")
|
||||||
|
|
||||||
|
# Add arguments for results per page
|
||||||
|
parser.add_argument('--results_per_page', type=int, default=100, help="Number of results per page (default: 100)")
|
||||||
|
|
||||||
|
# Parse the arguments
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
return args.results_per_page
|
||||||
|
|
||||||
|
def generate_token(username, password, url="https://www.arcgis.com/sharing/rest/generateToken"):
|
||||||
|
"""Generates an authentication token."""
|
||||||
|
payload = {
|
||||||
|
'f': 'json',
|
||||||
|
'username': username,
|
||||||
|
'password': password,
|
||||||
|
'client': 'referer',
|
||||||
|
'referer': 'https://www.arcgis.com',
|
||||||
|
'expiration': '120'
|
||||||
|
}
|
||||||
|
headers = {}
|
||||||
|
try:
|
||||||
|
response = requests.post(url, headers=headers, data=payload)
|
||||||
|
response.raise_for_status() # Raise an error for bad status codes
|
||||||
|
token = response.json()['token']
|
||||||
|
logger.info("Token generated successfully.")
|
||||||
|
return token
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
logger.error(f"Error generating token: {e}")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
def truncate(token, hostname, instance, fs, layer, secure=True):
|
||||||
|
"""Truncate the specified layer in the feature service."""
|
||||||
|
|
||||||
|
protocol = 'https://' if secure else 'http://'
|
||||||
|
url = f"{protocol}{hostname}/{instance}/arcgis/rest/admin/services/{fs}/FeatureServer/{layer}/truncate?token={token}&async=true&f=json"
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Attempt the POST request
|
||||||
|
logging.info(f"Attempting to truncate layer {layer} on {hostname}...")
|
||||||
|
response = requests.post(url, timeout=30)
|
||||||
|
|
||||||
|
# Check for HTTP errors
|
||||||
|
response.raise_for_status() # Raise an exception for HTTP errors (4xx, 5xx)
|
||||||
|
|
||||||
|
# Check for any known error in the response content (e.g., ArcGIS error codes)
|
||||||
|
if response.status_code == 200:
|
||||||
|
result = response.json()
|
||||||
|
if 'error' in result:
|
||||||
|
logging.error(f"Error truncating layer: {result['error']}")
|
||||||
|
return None
|
||||||
|
logging.info(f"Successfully truncated layer: {protocol}{hostname}/{instance}/arcgis/rest/admin/services/{fs}/FeatureServer/{layer}.")
|
||||||
|
return result
|
||||||
|
else:
|
||||||
|
logging.error(f"Unexpected response: {response.status_code} - {response.text}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
# Catch network-related errors, timeouts, etc.
|
||||||
|
logging.error(f"Request failed: {e}")
|
||||||
|
return None
|
||||||
|
except Exception as e:
|
||||||
|
# Catch any other unexpected errors
|
||||||
|
logging.error(f"An unexpected error occurred: {e}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
def add_features(token, hostname, instance, fs, layer, aggregated_data, secure=True):
|
||||||
|
"""Add features to a feature service."""
|
||||||
|
protocol = 'https://' if secure else 'http://'
|
||||||
|
url = f"{protocol}{hostname}/{instance}/arcgis/rest/services/{fs}/FeatureServer/{layer}/addFeatures?token={token}&rollbackOnFailure=true&f=json"
|
||||||
|
logger.info(f"Attempting to add features on {protocol}{hostname}/{instance}/arcgis/rest/services/{fs}/FeatureServer/{layer}...")
|
||||||
|
|
||||||
|
# Prepare features data as the payload
|
||||||
|
features_json = json.dumps(aggregated_data) # Convert aggregated data to JSON string
|
||||||
|
features_encoded = urllib.parse.quote(features_json) # URL-encode the JSON string
|
||||||
|
|
||||||
|
# Construct the URL-encoded payload with 'features=' and the URL-encoded data
|
||||||
|
payload = f"features={features_encoded}"
|
||||||
|
|
||||||
|
headers = {
|
||||||
|
'Content-Type': 'application/x-www-form-urlencoded'
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
response = requests.post(url, headers=headers, data=payload, timeout=180)
|
||||||
|
response.raise_for_status() # Raise an error for bad status codes
|
||||||
|
logger.info("Features added successfully.")
|
||||||
|
return response.json()
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
logger.error(f"Request error: {e}")
|
||||||
|
return {'error': str(e)}
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
logger.error(f"Error decoding JSON response: {e}")
|
||||||
|
return {'error': 'Invalid JSON response'}
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Main entry point for the script."""
|
||||||
|
# Parse command-line arguments
|
||||||
|
results_per_page = parse_arguments()
|
||||||
|
|
||||||
|
load_dotenv("753DataSync.env")
|
||||||
|
api_url = os.getenv('API_URL')
|
||||||
|
|
||||||
|
# Generate the token
|
||||||
|
username = os.getenv('AGOL_USER')
|
||||||
|
password = os.getenv('AGOL_PASSWORD')
|
||||||
|
token = generate_token(username, password)
|
||||||
|
|
||||||
|
# Set ArcGIS host details
|
||||||
|
hostname = os.getenv('HOSTNAME')
|
||||||
|
instance = os.getenv('INSTANCE')
|
||||||
|
fs = os.getenv('FS')
|
||||||
|
layer = os.getenv('LAYER')
|
||||||
|
|
||||||
|
# Truncate the layer before adding new features
|
||||||
|
truncate(token, hostname, instance, fs, layer)
|
||||||
|
|
||||||
|
all_data = []
|
||||||
|
page_number = 1
|
||||||
|
|
||||||
|
while True:
|
||||||
|
# Fetch data from the API
|
||||||
|
data = fetch_data(api_url, page_number, results_per_page)
|
||||||
|
|
||||||
|
# Append features data to the aggregated list
|
||||||
|
all_data.extend(data) # Data is now a list of features
|
||||||
|
|
||||||
|
# Generate filename with timestamp for the individual page
|
||||||
|
timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
|
||||||
|
page_filename = f"data/enforcement_page_{page_number}_results_{results_per_page}_{timestamp}.json"
|
||||||
|
|
||||||
|
# Save individual page data
|
||||||
|
save_json(data, page_filename)
|
||||||
|
|
||||||
|
# Check if the number of records is less than the results_per_page, indicating last page
|
||||||
|
if len(data) < results_per_page:
|
||||||
|
logger.info("No more data to fetch, stopping pagination.")
|
||||||
|
break
|
||||||
|
|
||||||
|
page_number += 1
|
||||||
|
|
||||||
|
# Prepare aggregated data
|
||||||
|
aggregated_data = all_data # Just use the collected features directly
|
||||||
|
|
||||||
|
# Save aggregated data to a single JSON file
|
||||||
|
aggregated_filename = f"data/aggregated_enforcement_results_{timestamp}.json"
|
||||||
|
save_json(aggregated_data, aggregated_filename)
|
||||||
|
|
||||||
|
# Add the features to the feature layer
|
||||||
|
response = add_features(token, hostname, instance, fs, layer, aggregated_data)
|
||||||
|
logger.info(f"Add features response: {json.dumps(response, indent=2)}")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
2
requirements.txt
Normal file
2
requirements.txt
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
requests==2.28.1
|
||||||
|
python-dotenv==1.0.0
|
||||||
Loading…
Reference in New Issue
Block a user