Create python CLI to generate csv for subscribed and onboarded laconicd accounts
This commit is contained in:
parent
9df6a71145
commit
62ece58a96
6
.gitignore
vendored
Normal file
6
.gitignore
vendored
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
venv
|
||||||
|
laconic_testnet.egg-info
|
||||||
|
__pycache__
|
||||||
|
|
||||||
|
*-deployment
|
||||||
|
*-spec.yml
|
32
cli/README.md
Normal file
32
cli/README.md
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
# cli
|
||||||
|
|
||||||
|
## Setup
|
||||||
|
|
||||||
|
* Run commands in repo root directory
|
||||||
|
```bash
|
||||||
|
python3 -m venv venv
|
||||||
|
source ./venv/bin/activate
|
||||||
|
```
|
||||||
|
|
||||||
|
* Install CLI in editable mode
|
||||||
|
```bash
|
||||||
|
pip install --editable .
|
||||||
|
```
|
||||||
|
|
||||||
|
* Verify installation
|
||||||
|
```bash
|
||||||
|
laconic-testnet --help
|
||||||
|
```
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
```bash
|
||||||
|
laconic-testnet --onboarded-json path/to/onboarded-accounts.json --csv-input path/to/subscribers.csv --output path/to/subscribed-onboarded-accounts.csv
|
||||||
|
```
|
||||||
|
|
||||||
|
## Cleanup
|
||||||
|
|
||||||
|
Deactivate virtual environment
|
||||||
|
```bash
|
||||||
|
deactivate
|
||||||
|
```
|
15
cli/cli.py
Normal file
15
cli/cli.py
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
import click
|
||||||
|
from .core import process_subscribers
|
||||||
|
|
||||||
|
@click.command()
|
||||||
|
@click.option('--onboarded-json', required=True, type=click.Path(exists=True), help='Path to onboarded accounts JSON file.')
|
||||||
|
@click.option('--subscribers-csv', required=True, type=click.Path(exists=True), help='Path to the subscribers CSV file.')
|
||||||
|
@click.option('--output', required=True, type=click.Path(), help='Path to the output CSV file.')
|
||||||
|
def main(onboarded_json, subscribers_csv, output):
|
||||||
|
"""
|
||||||
|
CLI tool to match subscriber data with participant data and generate a CSV.
|
||||||
|
"""
|
||||||
|
process_subscribers(onboarded_json, subscribers_csv, output)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
50
cli/core.py
Normal file
50
cli/core.py
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
import csv
|
||||||
|
import hashlib
|
||||||
|
import json
|
||||||
|
|
||||||
|
def hash_subscriber_id(subscriber_id):
|
||||||
|
return hashlib.sha256(subscriber_id.encode()).hexdigest()
|
||||||
|
|
||||||
|
def process_subscribers(onboarded_json, subscribers_csv, output):
|
||||||
|
# Load JSON data from the file
|
||||||
|
with open(onboarded_json, 'r') as json_file:
|
||||||
|
json_data = json.load(json_file)
|
||||||
|
|
||||||
|
# Create a dictionary mapping kyc_id to participant data
|
||||||
|
kyc_map = {participant['kyc_id']: participant for participant in json_data['participants']}
|
||||||
|
|
||||||
|
# Load subscribers data from the CSV file and process it using map
|
||||||
|
with open(subscribers_csv, 'r') as csv_file:
|
||||||
|
csv_reader = csv.DictReader(csv_file)
|
||||||
|
subscribers = list(csv_reader)
|
||||||
|
|
||||||
|
# Use map to process subscribers
|
||||||
|
def process_subscriber(subscriber):
|
||||||
|
hashed_subscriber_id = hash_subscriber_id(subscriber['subscriber_id'])
|
||||||
|
participant = kyc_map.get(hashed_subscriber_id)
|
||||||
|
|
||||||
|
if participant:
|
||||||
|
return {
|
||||||
|
'subscriber_id': subscriber['subscriber_id'],
|
||||||
|
'email': subscriber['email'],
|
||||||
|
'cosmos_address': participant['cosmos_address'],
|
||||||
|
'nitro_address': participant['nitro_address'],
|
||||||
|
'role': participant['role'],
|
||||||
|
'kyc_id': participant['kyc_id'],
|
||||||
|
'status': subscriber['status'],
|
||||||
|
'premium': subscriber['premium?'],
|
||||||
|
'created_at': subscriber['created_at']
|
||||||
|
}
|
||||||
|
return None
|
||||||
|
|
||||||
|
# Apply the map function and filter out None values
|
||||||
|
output_data = list(filter(None, map(process_subscriber, subscribers)))
|
||||||
|
|
||||||
|
# Write the matched data to a new CSV file
|
||||||
|
with open(output, 'w', newline='') as csv_file:
|
||||||
|
fieldnames = ['subscriber_id', 'email', 'cosmos_address', 'nitro_address', 'role', 'kyc_id', 'status', 'premium', 'created_at']
|
||||||
|
csv_writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
|
||||||
|
csv_writer.writeheader()
|
||||||
|
csv_writer.writerows(output_data)
|
||||||
|
|
||||||
|
print(f'Data has been written to {output}')
|
1
requirements.txt
Normal file
1
requirements.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
Click
|
Loading…
Reference in New Issue
Block a user