example.py

import os
import time
import logging

import stomp
import sqlite3
import pandas as pd

from topicmapping import TopicMapping
from datafeeder import RailDataFeeder

pd.options.display.max_columns = None

# Set up the personal information for the data feeds
USERNAME = 'XXXXXX'
PASSWORD = 'XXXXXX'

# four topics to choose from - 1. MVT 2. PPM 3. VSTP 4. TD
TOPIC = "MVT"

train_rdf = RailDataFeeder(
                    db_name=TopicMapping[TOPIC][2], 
                    channel=TopicMapping[TOPIC][1], 
                    topic=TOPIC,
                    schema=TopicMapping[TOPIC][0],
                    username=USERNAME,
                    password=PASSWORD,
                    drop_if_exists=True,
                    view=False
)

train_rdf.download_feed()

# convert to dataframe
train_rdf.to_pandas()