[859] | 1 | # -*- coding: ISO-8859-1 -*- |
---|
| 2 | |
---|
| 3 | ################################## |
---|
| 4 | # @program smon |
---|
| 5 | # @description simulation monitor |
---|
| 6 | # @copyright Copyright â(c)2009 Centre National de la Recherche Scientifique CNRS. |
---|
| 7 | # All Rights Reservedâ |
---|
| 8 | # @svn_file $Id: repo_io.py 2599 2013-03-24 19:01:23Z jripsl $ |
---|
| 9 | # @version $Rev: 2599 $ |
---|
| 10 | # @lastrevision $Date: 2013-03-24 20:01:23 +0100 (Sun, 24 Mar 2013) $ |
---|
| 11 | # @license CeCILL (http://dods.ipsl.jussieu.fr/jripsl/smon/LICENSE) |
---|
| 12 | ################################## |
---|
| 13 | |
---|
| 14 | """ |
---|
| 15 | This module contains repository I/O code |
---|
| 16 | """ |
---|
| 17 | |
---|
| 18 | import sys |
---|
| 19 | |
---|
| 20 | # line below is to include Prodiguer database I/O library in the search path |
---|
| 21 | sys.path.append("/home/jripsl/snapshot/src") |
---|
| 22 | |
---|
| 23 | # import Prodiguer database I/O library |
---|
| 24 | import elixir |
---|
| 25 | import prodiguer_shared |
---|
| 26 | |
---|
[865] | 27 | |
---|
| 28 | |
---|
| 29 | |
---|
| 30 | |
---|
[859] | 31 | |
---|
| 32 | |
---|
[871] | 33 | # --- module static initialization --- # |
---|
| 34 | |
---|
| 35 | CSTE_MODE_LOCAL_REPO="local_repo" |
---|
| 36 | CSTE_MODE_REMOTE_REPO="remote_repo" |
---|
| 37 | CSTE_MODE_REMOTE_REPO_STUB="remote_repo_stub" |
---|
| 38 | |
---|
| 39 | # set mode |
---|
| 40 | mode=CSTE_MODE_LOCAL_REPO # CSTE_MODE_LOCAL_REPO, CSTE_MODE_REMOTE_REPO, CSTE_MODE_REMOTE_REPO_STUB |
---|
| 41 | |
---|
| 42 | # set repository driver |
---|
| 43 | if mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 44 | import prodiguer_shared.repo.mq.hooks_stub as repo |
---|
| 45 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 46 | import prodiguer_shared.repo.mq.hooks as repo |
---|
| 47 | elif mode==CSTE_MODE_LOCAL_REPO: |
---|
| 48 | import local_repo as repo |
---|
| 49 | else: |
---|
| 50 | raise Exception("ERR001 - incorrect mode") |
---|
| 51 | |
---|
| 52 | |
---|
| 53 | |
---|
| 54 | |
---|
| 55 | |
---|
| 56 | |
---|
[865] | 57 | # -- methods -- # |
---|
[859] | 58 | |
---|
| 59 | def init(): |
---|
[865] | 60 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
[866] | 61 | repo.connect() |
---|
[865] | 62 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 63 | _CONNECTION = "postgresql://postgres:Silence107!@localhost:5432/prodiguer" |
---|
[859] | 64 | prodiguer_shared.connect(_CONNECTION) |
---|
[865] | 65 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
[859] | 66 | pass |
---|
| 67 | else: |
---|
| 68 | raise Exception("ERR004 - incorrect mode") |
---|
| 69 | |
---|
[865] | 70 | def free(): |
---|
| 71 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
[871] | 72 | repo.free() |
---|
[865] | 73 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 74 | |
---|
| 75 | #prodiguer_shared.close() |
---|
| 76 | pass |
---|
| 77 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 78 | pass |
---|
| 79 | else: |
---|
| 80 | raise Exception("ERR009 - incorrect mode") |
---|
| 81 | |
---|
[871] | 82 | def populate_tables_with_sample(): |
---|
| 83 | repo.populate_tables_with_sample() |
---|
| 84 | |
---|
| 85 | def retrieve_simulations(): |
---|
| 86 | return repo.retrieve_simulations() |
---|
| 87 | |
---|
[859] | 88 | def test(): |
---|
[865] | 89 | repo.create_message("test2", 2, "bla2") |
---|
[859] | 90 | commit() |
---|
| 91 | |
---|
[865] | 92 | repo.update_simulation_status('1pctCO22', 'ERROR') |
---|
[859] | 93 | commit() |
---|
| 94 | |
---|
[865] | 95 | repo.create_message("test3", 3, "bla3") |
---|
[859] | 96 | rollback() |
---|
| 97 | |
---|
| 98 | def commit(): |
---|
[865] | 99 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
[866] | 100 | repo.commit() |
---|
[865] | 101 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
[859] | 102 | elixir.session.commit() |
---|
[865] | 103 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
[859] | 104 | pass |
---|
| 105 | else: |
---|
| 106 | raise Exception("ERR002 - incorrect mode") |
---|
| 107 | |
---|
| 108 | def rollback(): |
---|
[865] | 109 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
[866] | 110 | repo.rollback() |
---|
[865] | 111 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
[859] | 112 | elixir.session.rollback() |
---|
[865] | 113 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
[859] | 114 | pass |
---|
| 115 | else: |
---|
| 116 | raise Exception("ERR003 - incorrect mode") |
---|
| 117 | |
---|
| 118 | def retrieve_simulation(name): |
---|
[865] | 119 | simulation=None |
---|
[859] | 120 | |
---|
[865] | 121 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 122 | simulation=repo.retrieve_simulation(name) |
---|
| 123 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 124 | |
---|
| 125 | # prepare args |
---|
| 126 | # .. |
---|
| 127 | |
---|
| 128 | # execute |
---|
| 129 | s=repo.retrieve_simulation(name) |
---|
| 130 | |
---|
| 131 | # process return values |
---|
| 132 | simulation=smon.types.Simulation(exec_start_date=s.ExecutionStartDate,exec_end_date=s.ExecutionEndDate,status=s.ExecutionState) # ExecutionState example: EXECUTION_STATE_RUNNING, EXECUTION_STATE_SET.. |
---|
| 133 | |
---|
| 134 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 135 | pass |
---|
| 136 | else: |
---|
| 137 | raise Exception("ERR014 - incorrect mode") |
---|
| 138 | |
---|
[866] | 139 | return simulation |
---|
[865] | 140 | |
---|
[859] | 141 | def delete_simulation(name): |
---|
[865] | 142 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 143 | repo.delete_simulation(name) |
---|
| 144 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
[859] | 145 | |
---|
[865] | 146 | # prepare args |
---|
| 147 | # .. |
---|
| 148 | |
---|
| 149 | # execute |
---|
| 150 | repo.delete_simulation(name) |
---|
| 151 | |
---|
| 152 | # process return values |
---|
| 153 | # .. |
---|
| 154 | |
---|
| 155 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 156 | pass |
---|
| 157 | else: |
---|
| 158 | raise Exception("ERR015 - incorrect mode") |
---|
| 159 | |
---|
[859] | 160 | def create_simulation(simulation): |
---|
[865] | 161 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 162 | repo.create_simulation(simulation) |
---|
| 163 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
[859] | 164 | |
---|
[865] | 165 | # prepare args |
---|
| 166 | # .. |
---|
[859] | 167 | |
---|
[865] | 168 | # execute |
---|
| 169 | repo.create_simulation(activity) |
---|
[859] | 170 | |
---|
[865] | 171 | # process return values |
---|
| 172 | # .. |
---|
[859] | 173 | |
---|
[865] | 174 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 175 | pass |
---|
| 176 | else: |
---|
| 177 | raise Exception("ERR016 - incorrect mode") |
---|
[859] | 178 | |
---|
[865] | 179 | def update_simulation_status(simulation): |
---|
| 180 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 181 | repo.update_simulation_status(simulation) |
---|
| 182 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
[859] | 183 | |
---|
[865] | 184 | # prepare args |
---|
| 185 | # .. |
---|
[859] | 186 | |
---|
[865] | 187 | # execute |
---|
| 188 | repo.update_simulation_status(name) |
---|
| 189 | |
---|
| 190 | # process return values |
---|
| 191 | # .. |
---|
| 192 | |
---|
| 193 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 194 | pass |
---|
| 195 | else: |
---|
| 196 | raise Exception("ERR017 - incorrect mode") |
---|
| 197 | |
---|
| 198 | def retrieve_messages(simulation): |
---|
| 199 | message=None |
---|
| 200 | |
---|
| 201 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 202 | message=repo.retrieve_messages(simulation) |
---|
| 203 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 204 | |
---|
| 205 | # prepare args |
---|
| 206 | # .. |
---|
| 207 | |
---|
| 208 | # execute |
---|
| 209 | repo.retrieve_messages(name) |
---|
| 210 | |
---|
| 211 | # process return values |
---|
| 212 | # .. |
---|
| 213 | |
---|
| 214 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 215 | pass |
---|
| 216 | else: |
---|
| 217 | raise Exception("ERR018 - incorrect mode") |
---|
| 218 | |
---|
| 219 | return message |
---|
| 220 | |
---|
| 221 | def delete_messages(simulation): |
---|
| 222 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 223 | repo.delete_messages(name) |
---|
| 224 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 225 | |
---|
| 226 | # prepare args |
---|
| 227 | # .. |
---|
| 228 | |
---|
| 229 | # execute |
---|
| 230 | repo.delete_messages(name) |
---|
| 231 | |
---|
| 232 | # process return values |
---|
| 233 | # .. |
---|
| 234 | |
---|
| 235 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 236 | pass |
---|
| 237 | else: |
---|
| 238 | raise Exception("ERR019 - incorrect mode") |
---|
| 239 | |
---|
| 240 | def create_message(message): |
---|
| 241 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 242 | repo.create_message(message) |
---|
| 243 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 244 | |
---|
| 245 | # prepare args |
---|
| 246 | # .. |
---|
| 247 | |
---|
| 248 | # execute |
---|
| 249 | repo.create_message() |
---|
| 250 | |
---|
| 251 | # process return values |
---|
| 252 | # .. |
---|
| 253 | |
---|
| 254 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 255 | pass |
---|
| 256 | else: |
---|
| 257 | raise Exception("ERR020 - incorrect mode") |
---|
| 258 | |
---|
| 259 | def retrieve_last_message(simulation): |
---|
| 260 | message=None |
---|
| 261 | |
---|
| 262 | if mode==CSTE_MODE_LOCAL_REPO: |
---|
| 263 | message=repo.retrieve_last_message(simulation) |
---|
| 264 | elif mode==CSTE_MODE_REMOTE_REPO: |
---|
| 265 | |
---|
| 266 | # prepare args |
---|
| 267 | # .. |
---|
| 268 | |
---|
| 269 | # execute |
---|
| 270 | repo.retrieve_last_message(simulation) |
---|
| 271 | |
---|
| 272 | # process return values |
---|
| 273 | # .. |
---|
| 274 | |
---|
| 275 | elif mode==CSTE_MODE_REMOTE_REPO_STUB: |
---|
| 276 | pass |
---|
| 277 | else: |
---|
| 278 | raise Exception("ERR021 - incorrect mode") |
---|
| 279 | |
---|
| 280 | return message |
---|
| 281 | |
---|
| 282 | |
---|
[859] | 283 | # --- higher level methods --- # |
---|
| 284 | |
---|
| 285 | def get_running_simulations(): |
---|
| 286 | running_simulation=[] |
---|
| 287 | |
---|
| 288 | for s in retrieve_simulation(): |
---|
| 289 | if s.status=="running": |
---|
| 290 | running_simulation.append(s) |
---|
| 291 | |
---|
| 292 | return running_simulation |
---|
[866] | 293 | |
---|
| 294 | |
---|