aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/sfIsisImporter.class.php21
1 files changed, 8 insertions, 13 deletions
diff --git a/lib/sfIsisImporter.class.php b/lib/sfIsisImporter.class.php
index 9233f64..d3866bf 100644
--- a/lib/sfIsisImporter.class.php
+++ b/lib/sfIsisImporter.class.php
@@ -72,22 +72,14 @@ class sfIsisImporter extends sfIsisImporterBase
}
/**
- * Set the primary key for the model by getting it or saving the model.
+ * Set the primary key for the model by getting it or just saving it.
*
* @param object $model Base model
*/
public function setBaseModelId(&$model)
{
- $id = $this->getBaseModelId($model);
-
- if ($id)
- {
- $model->id = $id;
- }
- else
- {
- $model->save();
- }
+ $model->id = $this->getBaseModelId($model);
+ $model->save();
}
/**
@@ -95,11 +87,14 @@ class sfIsisImporter extends sfIsisImporterBase
*
* @param object $model Base model
* @return int Base model id
- * @todo Write and test
*/
public function getBaseModelId(&$model)
{
- return false;
+ $method = get . $this->getModelName($model) .'PrimaryKey';
+ if (method_exists($this, $method))
+ {
+ return $this->{$method}($model);
+ }
}
/**