update_proposals.rkt 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. #lang racket/base
  2. (require racket/cmdline)
  3. (require racket/date)
  4. (require db)
  5. (define progname "update_proposals.rkt")
  6. ; load configuration file
  7. (require (file "config.rkt"))
  8. (date-display-format 'iso-8601)
  9. ; set up command line arguments
  10. (define mode (command-line
  11. #:program "update_proposals"
  12. #:args (updatetype) ; (add, update)
  13. updatetype))
  14. ; print some help
  15. (define (printhelp)
  16. (write-string (string-append "Usage: "
  17. progname " MODE\n\n"))
  18. (write-string "Where MODE is one of:\n")
  19. (write-string " add\t\t - add new proposal to database.\n")
  20. (write-string " update\t\t - update a a proposal with results.\n")
  21. (write-string " help\t\t - Show this help message.\n")
  22. (write-string "\nCopyright 2019 George C. Privon\n"))
  23. ; set up a condensed prompt for getting information
  24. (define (getinput prompt)
  25. (write-string prompt)
  26. (write-string ": ")
  27. (read-line))
  28. ; take an input result from the SQL search and write it out nicely
  29. (define (printentry entry)
  30. (write-string (string-append
  31. (number->string (vector-ref entry 0))
  32. ": "
  33. (vector-ref entry 1)
  34. "("
  35. (vector-ref entry 2)
  36. ") "
  37. (vector-ref entry 3)
  38. ".\n"))
  39. )
  40. (define (addnew)
  41. (write-string "Adding new proposal to database.\n")
  42. ; user inputs proposal data
  43. (define proptype (getinput "Proposal type"))
  44. (define org (getinput "Submitting organization"))
  45. (define solic (getinput "Solitation/call"))
  46. (define tele (getinput "Telescope"))
  47. (define title (getinput "Proposal title"))
  48. (define pi (getinput "PI"))
  49. (define coi (getinput "CoIs"))
  50. ; assume all these proposals are submitted, don't ask the user
  51. (define status "submitted")
  52. (define submitdate (getinput "Submit date"))
  53. (define oID (getinput "Organization's proposal ID"))
  54. ; do the INSERT into the Sqlite database
  55. (query-exec conn "INSERT INTO proposals (type, organization, solicitation, telescope, PI, title, CoI, status, submitdate, orgpropID) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
  56. proptype org solic tele pi title coi status submitdate oID))
  57. (define (update ID)
  58. (write-string (string-append "Updating entry " (number->string ID) "\n"))
  59. (define entry (query-row conn "SELECT * FROM proposals WHERE ID=?" ID))
  60. (write-string (string-append "Current status is: "
  61. (vector-ref entry 9)
  62. " ("
  63. (vector-ref entry 10)
  64. ")\n"))
  65. (write-string "Please enter new status: ")
  66. (define newstatus (read-line))
  67. ;(write-string "Please enter date of updated status (leave blank to use current date): ")
  68. ;(define resdate (read-line))
  69. (define resdate (date->string (seconds->date (current-seconds))))
  70. ; now update that entry
  71. (query-exec conn "UPDATE proposals SET status=?, resultdate=? WHERE ID=?"
  72. newstatus
  73. resdate
  74. ID)
  75. (write-string "Entry updated.\n")
  76. )
  77. ; find proposals waiting for updates
  78. (define (findpending)
  79. (write-string "Updating proposals")
  80. ; retrieve all proposals whose status is still listed as "submitted"
  81. (define unfinished (query-rows conn "SELECT ID,telescope,solicitation,title FROM proposals WHERE status='submitted'"))
  82. (write-string (string-append (make-string (length unfinished)) " pending proposals found:\n"))
  83. (map printentry unfinished)
  84. (write-string "Please enter a proposal number to edit (enter 0 or nothing to exit): ")
  85. (define upID (read-line))
  86. (cond
  87. [(eq? (string->number upID) 0) (exit)]
  88. [(string->number upID) (update (string->number upID))]
  89. [else (exit)])
  90. )
  91. ; make sure we can use the sqlite3 connection
  92. (cond (not (sqlite3-available?))
  93. (error "Sqlite3 library not available."))
  94. ; open the database file
  95. (define conn (sqlite3-connect #:database dbloc))
  96. ; determine which mode we're in
  97. (cond
  98. [(regexp-match "help" mode) (printhelp)]
  99. [(regexp-match "add" mode) (addnew)]
  100. [(regexp-match "update" mode) (findpending)])
  101. ; close the databse
  102. (disconnect conn)