Add missing user environment variables (#51)

* debug multi user

Signed-off-by: vsoch <[email protected]>

CHANGELOG.md

@@ -14,6 +14,7 @@ and **Merged pull requests**. Critical items to know are:
 The versions coincide with releases on pip. Only major versions will be released as tags on Github.
 
 ## [0.0.x](https://github.com/flux-framework/flux-restful-api/tree/main) (0.0.x)
+ - Ensure we update flux environment for user (0.1.13)
  - Add better multi-user mode - running jobs on behalf of user (0.1.12)
  - Restore original rpc to get job info (has more information) (0.1.11)
  - Refactor of FLux Restful to use a database and OAauth2 (0.1.0)

VERSION

@@ -1 +1 @@
-0.1.12
+0.1.13

app/library/flux.py

@@ -1,5 +1,6 @@
 import json
 import os
+import pwd
 import re
 import shlex
 import subprocess
@@ -37,10 +38,16 @@ def submit_job(handle, fluxjob, user):
         print("Submit in single-user mode.")
         return flux.job.submit_async(handle, fluxjob)
 
+    pw_record = pwd.getpwnam(user)
+    user_name = pw_record.pw_name
+    # user_uid = pw_record.pw_uid
+    # user_gid = pw_record.pw_gid
+
     # Update the payload for the correct user
-    # Use helper script to sign payload
+    fluxjob.environment["HOME"] = pw_record.pw_dir
+    fluxjob.environment["LOGNAME"] = user_name
+    fluxjob.environment["USER"] = pw_record.pw_name
     payload = json.dumps(fluxjob.jobspec)
-    # payload['HOME'] =
 
     # We ideally need to pipe the payload into flux python
     try:

clients/python/flux_restful_client/version.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.12"
+__version__ = "0.1.13"
 AUTHOR = "Vanessa Sochat"
 EMAIL = "[email protected]"
 NAME = "flux-restful-client"