Simple HTML Grid widget – part 4

I know that last time I said we would look at passing data to other pages and data sorting but instead I thought I would make a brief diversion into sqlite. Playing around with table data such as this:

a	b	c
1	2	3
4	5	6
7	8	9

which is rendered by:

(print-ln (render-table (make-table
                         '("a" "b" "c")
                         '(("1" "2" "3")
                           ("4" "5" "6")
                           ("7" "8" "9")))
                        '()))

is pretty uninspiring. Instead, I can use some financial data that is available from yahoo. Data for the first 50 (or so) NASDAQ stocks is available here.

(require (lib "url.ss" "net"))

(define (print-ln . args)
  (for-each display args)
  (newline))

(define *url-prefix* "http://download.finance.yahoo.com/d/quotes.csv?")
(define *url* (string-append *url-prefix*
                             "s=@%5EIXIC&f=sl1d1t1c1ohgv&e=.csv&h=50"))

(define (display-url url)
  (let ((p (get-pure-port (string->url url))))
    (let loop ((line (read-line p)))
      (unless (eof-object? line)
        (print-ln line)
        (loop (read-line p))))
    (close-input-port p)))

(display-url *url*)

Running this script gives the rather strange error message Missing Format Variable.

(url->string (string->url *url*))

reveals the reason. string->url corrupts the url! It should be this:
http://download.finance.yahoo.com/d/quotes.csv?s=@%5EIXIC&f=sl1d1t1c1ohgv&e=.csv&h=50

But instead it is this (ampersands converted to semi-colons amongst aother things):
http://download.finance.yahoo.com/d/quotes.csv?s=%40%5EIXIC;f=sl1d1t1c1ohgv;e=.csv;h=50

Yahoo rejects the later url with the error message as above.

Absolutely unbe-fricking-lievable.

I can’t imagine the thinking behind making string->url take a perfectly valid URL and breaking it. If you must do this, at least provide a function that doesn’t have this behaviour, say string->url and string->borked-url. This totally unreasonable design decision has damaged my confidence in the PLT scheme standard library. It is possible to work around the stupid default behaviour (although the fix presumably breaks URLs with semi-colons in the query string).

(require (lib "url.ss" "net")
         (lib "uri-codec.ss" "net"))

(current-alist-separator-mode 'amp)

(define (print-ln . args)
  (for-each display args)
  (newline))

(define *url-prefix* "http://download.finance.yahoo.com/d/quotes.csv?")
(define *url* (string-append *url-prefix*
                             "s=@%5EIXIC&f=sl1d1t1c1ohgv&e=.csv&h=50"))

(define (display-url url)
  (let ((p (get-pure-port (string->url url))))
    (let loop ((line (read-line p)))
      (unless (eof-object? line)
        (print-ln line)
        (loop (read-line p))))
    (close-input-port p)))

(display-url *url*)

The next step is to store the data in the database. First create a suitable table:

CREATE TABLE stock_data
(
    ticker varchar(8),
    price double,
    change double,
    date datetime
)

There is a nice csv library on planet. You declare a csv reader like this:

(define make-csv-reader
  (make-csv-reader-maker
   '((separator-chars            . (#\,))
     (strip-leading-whitespace?  . #t)
     (strip-trailing-whitespace? . #t))))

Then the functions to extract the interesting data and store it in the database. The symbol, price and change from the previous day are the first, second and fifth fields respectively. I think this csv file only changes once a day so we fix the time portion of the datetime to an arbitrary value. I haven’t named the functions particularly well here, but they should convey my intent at least.

(define (db-insert-row symbol price change)
  (let ((sql (format "
    INSERT INTO stock_data
    (ticker, price, change, date)
    VALUES
    ('~a', ~a, ~a, strftime('%Y-%m-%d 06:00:00', 'now'))
    " symbol price change)))
    (print-ln sql)
    (exec/ignore *dbh* sql)))

(define (csv-insert-into-db p)
  (let ((reader (make-csv-reader p)))
    (let loop ()
      (let ((l (reader)))
        (unless (or (null? l)
                    (null? (cdr l)))
          (let ((symbol (first l))
                (price (second l))
                (change (fifth l)))
            (db-insert-row symbol price change))
          (loop))))))

And here is the complete script:

(require (lib "1.ss" "srfi")
         (lib "url.ss" "net")
         (lib "uri-codec.ss" "net")
         (planet "csv.ss" ("neil" "csv.plt" 1 1))
         (planet "sqlite.ss" ("jaymccarthy" "sqlite.plt" 3 1)))

(current-alist-separator-mode 'amp)

(define (print-ln . args)
  (for-each display args)
  (newline))

(define *url-prefix* "http://download.finance.yahoo.com/d/quotes.csv?")
(define *url* (string-append *url-prefix*
                             "s=@%5EIXIC&f=sl1d1t1c1ohgv&e=.csv&h=50"))

(define (display-url url)
  (let ((p (get-pure-port (string->url url))))
    (let loop ((line (read-line p)))
      (unless (eof-object? line)
        (print-ln line)
        (loop (read-line p))))
    (close-input-port p)))

(define make-csv-reader
  (make-csv-reader-maker
   '((separator-chars            . (#\,))
     (strip-leading-whitespace?  . #t)
     (strip-trailing-whitespace? . #t))))

(define (db-insert-row symbol price change)
  (let ((sql (format "
    INSERT INTO stock_data
    (ticker, price, change, date)
    VALUES
    ('~a', ~a, ~a, strftime('%Y-%m-%d 06:00:00', 'now'))
    " symbol price change)))
    (print-ln sql)
    (exec/ignore *dbh* sql)))

(define (csv-insert-into-db p)
  (let ((reader (make-csv-reader p)))
    (let loop ()
      (let ((l (reader)))
        (unless (or (null? l)
                    (null? (cdr l)))
          (let ((symbol (first l))
                (price (second l))
                (change (fifth l)))
            (db-insert-row symbol price change))
          (loop))))))

(define (process-html url processor)
  (let ((p (get-pure-port (string->url url))))
    (processor p)
    (close-input-port p)))

(define *path-prefix* "c:/tmp")
(define (path p)
  (string->path (string-append *path-prefix* "/" p)))

(define *dbh* (open (path "test.db")))

(process-html *url* csv-insert-into-db)

(select *dbh* "SELECT count(*) as stocks from stock_data")
(define *results* (select *dbh* "SELECT * from stock_data"))

(for-each print-ln *results*)

(close *dbh*)

Advertisements