1 files changed, 51 insertions, 0 deletions
diff --git a/net/test/python_utils.cc b/net/test/python_utils.cc
new file mode 100644
index 0000000..5f574fe
--- /dev/null
+++ b/net/test/python_utils.cc
@@ -0,0 +1,51 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "net/test/python_utils.h"
+
+#include "base/base_paths.h"
+#include "base/environment.h"
+#include "base/file_path.h"
+#include "base/path_service.h"
+#include "base/scoped_ptr.h"
+#include "base/utf_string_conversions.h"
+
+const char kPythonPathEnv[] = "PYTHONPATH";
+
+void AppendToPythonPath(const FilePath& dir) {
+  scoped_ptr<base::Environment> env(base::Environment::Create());
+  std::string old_path;
+  std::string dir_path;
+#if defined(OS_WIN)
+  dir_path = WideToUTF8(dir.value());
+#elif defined(OS_POSIX)
+  dir_path = dir.value();
+#endif
+  if (!env->GetVar(kPythonPathEnv, &old_path)) {
+    env->SetVar(kPythonPathEnv, dir_path.c_str());
+  } else if (old_path.find(dir_path) == std::string::npos) {
+    std::string new_path(old_path);
+#if defined(OS_WIN)
+    new_path.append(";");
+#elif defined(OS_POSIX)
+    new_path.append(":");
+#endif
+    new_path.append(dir_path.c_str());
+    env->SetVar(kPythonPathEnv, new_path);
+  }
+}
+
+bool GetPythonRunTime(FilePath* dir) {
+#if defined(OS_WIN)
+  if (!PathService::Get(base::DIR_SOURCE_ROOT, dir))
+    return false;
+  *dir = FilePath(FILE_PATH_LITERAL("third_party"))
+      .Append(FILE_PATH_LITERAL("python_24"))
+      .Append(FILE_PATH_LITERAL("python.exe"));
+#elif defined(OS_POSIX)
+  *dir = FilePath("python");
+#endif
+  return true;
+}
+