Changeset 42132d7 for client

Timestamp:

Jul 9, 2014, 4:30:43 AM (11 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master

Children:

db5be5d

Parents:

4f2c77a

git-author:

Alex Dehnert <adehnert@…> (07/09/14 04:30:43)

git-committer:

Alex Dehnert <adehnert@…> (07/09/14 04:30:43)

Message:

Working RFP downloader and creator

Location:

client

Files:

• rfp.py (added)
• sapweb.py (modified) (3 diffs)

client/sapweb.py

@@ -2 +2 @@
 
 import csv
+import os
 import sys
+import urllib2
+import urlparse
 
 import selenium.common.exceptions
@@ -8 +11 @@
 import pysapweb.rfp
 import pysapweb.sap_profiles
+
+use_browser = False
+use_browser = True
 
 def get_profile():
@@ -35 +41 @@
     return id, name, address, line_item, payee
 
-def create_rfps(fd):
+def download_doc(url, docs_dir):
+    parsed_url = urlparse.urlparse(url)
+    basename = os.path.basename(parsed_url.path)
+    doc_fn = os.path.join(docs_dir, basename)
+    doc_data = urllib2.urlopen(url).read()
+    with open(doc_fn, 'w') as doc_fd:
+        doc_fd.write(doc_data)
+    return doc_fn
+
+def create_rfps(fd, docs_dir):
     reader = csv.DictReader(fd)
-    browser = get_profile()
+    if use_browser:
+        browser = get_profile()
+    created = []
     for line in reader:
         id, name, address, line_item, payee = read_rfp_info(line)
+        doc_url = line['documentation']
+        doc_path = download_doc(doc_url, docs_dir)
         args = dict(
             name=name,
             payee=payee,
             line_items=(line_item, ),
+            receipts=(doc_path, ),
         )
         if not payee[0]:
             args['address'] = address
-        print "args=%s" % (args, )
-        rfp = pysapweb.rfp.create(browser, **args)
+        if use_browser:
+            rfp = pysapweb.rfp.create(browser, **args)
+        else:
+            rfp = 123456
         print "Created voucher %d -> RFP %s" % (id, rfp)
+        line['rfp_number'] = rfp
+        created.append(line)
+    return created
 
 def create_test_rfp():